提交 cb0e84e0 authored 作者: Gijs van Tulder's avatar Gijs van Tulder

Make test_abstract_conv 3D tests a bit smaller/faster.

上级 10caddc2
......@@ -598,8 +598,8 @@ class BaseTestConv3d(BaseTestConv):
def setup_class(cls):
if theano.config.blas.ldflags == '':
raise SkipTest("BLAS required for reference")
cls.inputs_shapes = [(2, 1, 6, 6, 6), (2, 2, 7, 5, 6)]
cls.filters_shapes = [(3, 1, 2, 2, 2), (1, 2, 2, 3, 1)]
cls.inputs_shapes = [(2, 1, 5, 5, 5), (1, 2, 7, 5, 6)]
cls.filters_shapes = [(2, 1, 2, 2, 2), (1, 2, 2, 1, 3)]
cls.subsamples = [(1, 1, 1), (2, 2, 2), (1, 2, 3)]
cls.default_subsamples = (1, 1, 1)
cls.filters_dilations = [(1, 1, 1), (1, 2, 1), (2, 1, 2)]
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论