提交 85908603 authored 作者: James Bergstra's avatar James Bergstra

added possibility to ignore_errors in run_test_conv_nnet2_classif to benchmark…

added possibility to ignore_errors in run_test_conv_nnet2_classif to benchmark even broken implementations.
上级 dc678d1f
...@@ -279,7 +279,12 @@ def run_conv_nnet2_classif(shared_fn, isize, ksize): ...@@ -279,7 +279,12 @@ def run_conv_nnet2_classif(shared_fn, isize, ksize):
print_mode(mode) print_mode(mode)
return rval return rval
def run_test_conv_nnet2_classif(seed, isize, ksize): def run_test_conv_nnet2_classif(seed, isize, ksize, ignore_error=False):
if ignore_error:
numpy.random.seed(seed)
rval_gpu = run_conv_nnet2_classif(tcn.shared_constructor, isize, ksize)
return
numpy.random.seed(seed) numpy.random.seed(seed)
rval_cpu = run_conv_nnet2_classif(shared, isize, ksize) rval_cpu = run_conv_nnet2_classif(shared, isize, ksize)
numpy.random.seed(seed) numpy.random.seed(seed)
...@@ -290,7 +295,7 @@ def test_lenet_28(): #MNIST ...@@ -290,7 +295,7 @@ def test_lenet_28(): #MNIST
run_test_conv_nnet2_classif(23485, 28, 5) run_test_conv_nnet2_classif(23485, 28, 5)
def test_lenet_32(): #CIFAR10 / Shapeset def test_lenet_32(): #CIFAR10 / Shapeset
run_test_conv_nnet2_classif(23485, 32, 5) run_test_conv_nnet2_classif(23485, 32, 5, ignore_error=False)
def test_lenet_108(): # NORB def test_lenet_108(): # NORB
run_test_conv_nnet2_classif(23485, 108, 7) run_test_conv_nnet2_classif(23485, 108, 7)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论