提交 9132a2a6 authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Don't show scary errors during the abstractconv tests.

上级 ea1bd6b9
......@@ -663,7 +663,6 @@ int main() {
else:
check_force_gemv_init._force_init_beta = False
print("init beta:", check_force_gemv_init._force_init_beta)
return check_force_gemv_init._force_init_beta
check_force_gemv_init._force_init_beta = None
......
......@@ -280,6 +280,11 @@ class TestCpuConv2d(BaseTestConv2d):
def setUp(self):
super(TestCpuConv2d, self).setUp()
self.mode = theano.compile.mode.get_default_mode().excluding('conv_gemm')
self.opt_err = theano.config.on_opt_error
theano.config.on_opt_error = 'ignore'
def tearDown(self):
theano.config.on_opt_error = self.opt_err
def tcase(self, i, f, s, b, flip, provide_shape):
mode = self.mode
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论