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

put the acceptance threashold more strict for test of the gpu conv since we fixed some problem.

上级 c1463e24
...@@ -422,7 +422,7 @@ def cmp_run_conv_nnet2_classif(seed, isize, ksize, bsize, ...@@ -422,7 +422,7 @@ def cmp_run_conv_nnet2_classif(seed, isize, ksize, bsize,
theano.tensor.basic.float32_atol=orig_float32_atol theano.tensor.basic.float32_atol=orig_float32_atol
if not cpu_only: if not cpu_only:
if verbose or not numpy.allclose(rval_cpu, rval_gpu,rtol=1e-3,atol=float_atol): if verbose or not numpy.allclose(rval_cpu, rval_gpu,rtol=1e-5,atol=float_atol):
print "cpu:", rval_cpu print "cpu:", rval_cpu
print "gpu:", rval_gpu print "gpu:", rval_gpu
print "abs diff:", numpy.absolute(rval_gpu-rval_cpu) print "abs diff:", numpy.absolute(rval_gpu-rval_cpu)
...@@ -465,13 +465,13 @@ def test_lenet_64(): # ??? ...@@ -465,13 +465,13 @@ def test_lenet_64(): # ???
cmp_run_conv_nnet2_classif(23485, 64, 7, 10, n_train=10, cmp_run_conv_nnet2_classif(23485, 64, 7, 10, n_train=10,
ignore_error=ignore_error, gpu_only=gpu_only, ignore_error=ignore_error, gpu_only=gpu_only,
cpu_only=cpu_only, verbose=verbose, cpu_only=cpu_only, verbose=verbose,
float_atol=5e-4, check_isfinite=True, version=version) check_isfinite=True, version=version)
def test_lenet_108(): # NORB def test_lenet_108(): # NORB
cmp_run_conv_nnet2_classif(23485, 108, 7, 5, n_train=4, cmp_run_conv_nnet2_classif(23485, 108, 7, 5, n_train=4,
ignore_error=ignore_error, gpu_only=gpu_only, ignore_error=ignore_error, gpu_only=gpu_only,
cpu_only=cpu_only, verbose=verbose, cpu_only=cpu_only, verbose=verbose,
check_isfinite=True, version=version, float_atol=7e-2) check_isfinite=True, version=version)
def test_lenet_256(): # ImageNet def test_lenet_256(): # ImageNet
cmp_run_conv_nnet2_classif(23485, 256, 9, 2, n_train=5, cmp_run_conv_nnet2_classif(23485, 256, 9, 2, n_train=5,
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论