Unverified 提交 314f4e2f authored 作者: Frédéric Bastien's avatar Frédéric Bastien 提交者: GitHub

Try to fix the test

上级 674b22e0
...@@ -1330,7 +1330,8 @@ class TestBilinearUpsampling(unittest.TestCase): ...@@ -1330,7 +1330,8 @@ class TestBilinearUpsampling(unittest.TestCase):
x = np.random.rand(1, 1, 200, 200).astype(theano.config.floatX) x = np.random.rand(1, 1, 200, 200).astype(theano.config.floatX)
resize = (24, 20) resize = (24, 20)
z = bilinear_upsampling(tensor.as_tensor_variable(x), frac_ratio=resize, use_1D_kernel=False) z = bilinear_upsampling(tensor.as_tensor_variable(x), frac_ratio=resize, use_1D_kernel=False)
utt.assert_allclose(z.shape.eval(mode='FAST_RUN'), (1, 1, 240, 240)) out = theano.function([], z.shape, mode='FAST_RUN')()
utt.assert_allclose(out, (1, 1, 240, 240))
class TestConv2dTranspose(unittest.TestCase): class TestConv2dTranspose(unittest.TestCase):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论