def unittest(): data = gpuarray.to_gpu(np.zeros((1, 3, 448, 448), dtype=np.float32)) yolo = loadMiniYolo(None, numOutput=1470, initscheme="gaussian") yolo(data) del yolo memPool.freeHeld()
def unittest(): nin = loadNiNImageNet(None, initscheme="gaussian") data = gpuarray.to_gpu(np.random.randn(1, 3, 224, 224).astype(np.float32)) nin(data) del nin memPool.freeHeld()
def unittest(): inmaps, nlabels = 161, 29 w2l = loadW2L(None, inmaps=inmaps, nlabels=29) shape = (16, inmaps, 200) data = gpuarray.to_gpu(np.random.randn(*shape).astype(np.float32)) w2l(data) del w2l memPool.freeHeld()
def unittest(): vocabsize = 1000 sentlength, embsize = 100, 128 data = gpuarray.to_gpu( np.random.randint(0, vocabsize, (1, sentlength), dtype=np.int32)) senti = buildNet(vocabsize, (3, 5, 7), None, sentlength, embsize, 1.0) senti(data) del senti memPool.freeHeld()
def unittest(): bn = loadInceptionBN(None, initscheme="gaussian") data = gpuarray.to_gpu(np.random.randn(1, 3, 224, 224).astype(np.float32)) bn(data) del bn memPool.freeHeld() v3 = loadInceptionV3(None, initscheme="gaussian") data = gpuarray.to_gpu(np.random.randn(1, 3, 299, 299).astype(np.float32)) v3(data) del v3 memPool.freeHeld()
def unittest(): data = gpuarray.to_gpu(np.random.randn(1, 3, 224, 224).astype(np.float32)) res = loadResNet(None, layers="50", initscheme="gaussian") res(data) del res memPool.freeHeld() res = loadResNet(None, layers="101", initscheme="gaussian") res(data) del res memPool.freeHeld() res = loadResNet(None, layers="152", initscheme="gaussian") res(data) del res memPool.freeHeld()
def unittest(): data = gpuarray.to_gpu(np.random.randn(1, 3, 224, 224).astype(np.float32)) vgg11 = loadVGG(None, layers="11", initscheme="gaussian") vgg11(data) del vgg11 memPool.freeHeld() vgg16 = loadVGG(None, layers="16", initscheme="gaussian") vgg16(data) del vgg16 memPool.freeHeld() vgg19 = loadVGG(None, layers="19", initscheme="gaussian") vgg19(data) del vgg19 memPool.freeHeld()