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()