提交 69f0732b authored 作者: Yann N. Dauphin's avatar Yann N. Dauphin

fix test

上级 b66c5e7d
......@@ -312,9 +312,11 @@ def test_downsample():
ggf = gradient.Lop(tensor.grad((ds_op(
tensor.as_tensor_variable(a))**2).sum(), a), a, a)
ref_mode = copy.copy(theano.compile.get_default_mode())
ref_mode = copy.copy(mode_without_gpu)
ref_mode.check_py_code = False
gg = pfunc([], ggf, mode=ref_mode)
gpu_mode = copy.copy(mode_with_gpu)
gpu_mode.check_py_code = False
gg = pfunc([], ggf, mode=gpu_mode)
gg2 = pfunc([], ggf, mode=ref_mode)
assert any([isinstance(node.op,
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论