提交 5e010a37 authored 作者: Frederic Bastien's avatar Frederic Bastien

remove useless warning in test.

上级 2144cc39
...@@ -180,9 +180,8 @@ def test_downsample(): ...@@ -180,9 +180,8 @@ def test_downsample():
f.maker.env.toposort()]) f.maker.env.toposort()])
assert numpy.allclose(f(),f2()) assert numpy.allclose(f(),f2())
g = pfunc([], tensor.grad(ds_op(tensor.as_tensor_variable(a)),a), mode=mode_with_gpu) g = pfunc([], tensor.grad(ds_op(tensor.as_tensor_variable(a)).sum(),a), mode=mode_with_gpu)
g2 = pfunc([], tensor.grad(ds_op(tensor.as_tensor_variable(a)),a), mode=mode_without_gpu) g2 = pfunc([], tensor.grad(ds_op(tensor.as_tensor_variable(a)).sum(),a), mode=mode_without_gpu)
print g.maker.env.toposort()
assert any([isinstance(node.op, tcn.blas.GpuDownsampleFactorMaxGrad) assert any([isinstance(node.op, tcn.blas.GpuDownsampleFactorMaxGrad)
for node in g.maker.env.toposort()]) for node in g.maker.env.toposort()])
assert numpy.allclose(g(),g2()) assert numpy.allclose(g(),g2())
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论