提交 8a40a29e authored 作者: Frederic's avatar Frederic

Make a test faster

上级 f61e0787
......@@ -138,7 +138,7 @@ class TestSP(unittest.TestCase):
# fixed parameters
bsize = 10 # batch size
imshp = (28,28)
imshp = (8, 8)
kshp = (5,5)
nkern = 1 # per output pixel
ssizes = ((1,1),(2,2))
......@@ -151,8 +151,8 @@ class TestSP(unittest.TestCase):
rng = numpy.random.RandomState(3423489)
import theano.gof as gof
#Mode(optimizer='fast_run', linker=gof.OpWiseCLinker(allow_gc=False)),):
for mode in ('FAST_COMPILE','FAST_RUN'): #,profmode):
for mode in (None,):
ntot, ttot = 0,0
for conv_mode in convmodes:
for ss in ssizes:
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论