提交 15d5e390 authored 作者: Frederic Bastien's avatar Frederic Bastien

Fix a test in mode=FAST_COMPILE

上级 f66ea7e8
......@@ -63,7 +63,8 @@ class TestCorr3DMM(unittest.TestCase):
filter_dilation=filter_dilation,
subsample=subsample),
[inputs_val.transpose(0, 4, 1, 2, 3),
filters_val.transpose(0, 4, 1, 2, 3)])
filters_val.transpose(0, 4, 1, 2, 3)],
mode=mode_with_gpu)
def test_valid(self):
self.run_conv_valid(inputs_shape=(16, 20, 12, 16, 1),
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论