提交 3a3424c2 authored 作者: Frederic Bastien's avatar Frederic Bastien

fix test in FAST_COMPILE

上级 b5e86fdb
......@@ -385,7 +385,9 @@ def test_neibs_grad_verify_grad():
def fn(images):
return T.sum(T.sqr(images2neibs(images, (2,2))), axis=[0,1])
unittest_tools.verify_grad(fn, [images_val])
unittest_tools.verify_grad(fn, [images_val], mode=mode_without_gpu)
if cuda.cuda_available:
unittest_tools.verify_grad(fn, [images_val], mode=mode_with_gpu)
if __name__ == '__main__':
#test_neibs_gpu()
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论