提交 953261ed authored 作者: Frederic Bastien's avatar Frederic Bastien

lower precission needed for big mlp on the gpu.

上级 c8e9e0e5
......@@ -118,7 +118,7 @@ def test_run_nnet():
print "time cpu: %f, time gpu: %f, speed up %f"%(tc, tg, tc/tg)
rtol = 1e-4
if n_in*n_hid>=2048*4096:
rtol = 5e-4
rtol = 7e-4
if not numpy.allclose(rval_cpu, rval_gpu,rtol=1e-4,atol=1e-6):
assert numpy.allclose(rval_cpu, rval_gpu,rtol=rtol,atol=1e-6)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论