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

Make a test faster

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