提交 ab86c74a authored 作者: Vincent Dumoulin's avatar Vincent Dumoulin

Fix test failing in FAST_COMPILE mode

上级 35ab10ca
...@@ -1562,13 +1562,17 @@ class TestConv2dTranspose(unittest.TestCase): ...@@ -1562,13 +1562,17 @@ class TestConv2dTranspose(unittest.TestCase):
output shape. output shape.
""" """
mode = self.mode
if theano.config.mode == "FAST_COMPILE":
mode = mode.excluding("conv_gemm").excluding('AbstractConvCheck')
output = theano.function( output = theano.function(
inputs=[], inputs=[],
outputs=conv2d_transpose(input=tensor.ones((2, 2, 4, 4)), outputs=conv2d_transpose(input=tensor.ones((2, 2, 4, 4)),
filters=tensor.ones((2, 1, 4, 4)), filters=tensor.ones((2, 1, 4, 4)),
output_shape=(2, 1, 10, 10), output_shape=(2, 1, 10, 10),
input_dilation=(2, 2)), input_dilation=(2, 2)),
mode=self.mode)() mode=mode)()
expected_output = numpy.array( expected_output = numpy.array(
[[[[2, 2, 4, 4, 4, 4, 4, 4, 2, 2], [[[[2, 2, 4, 4, 4, 4, 4, 4, 2, 2],
[2, 2, 4, 4, 4, 4, 4, 4, 2, 2], [2, 2, 4, 4, 4, 4, 4, 4, 2, 2],
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论