提交 af91444c 编写于 作者: Y Yancey1989

polish unittest test=develop

上级 35cda13e
...@@ -115,9 +115,7 @@ class TestMNIST(TestParallelExecutorBase): ...@@ -115,9 +115,7 @@ class TestMNIST(TestParallelExecutorBase):
def test_simple_fc(self): def test_simple_fc(self):
# use_cuda # use_cuda
if core.is_compiled_with_cuda(): self.check_simple_fc_convergence(True)
self.check_simple_fc_convergence(True)
self.check_simple_fc_convergence(True, use_reduce=False)
self.check_simple_fc_convergence(False) self.check_simple_fc_convergence(False)
def test_simple_fc_with_new_strategy(self): def test_simple_fc_with_new_strategy(self):
...@@ -154,8 +152,7 @@ class TestMNIST(TestParallelExecutorBase): ...@@ -154,8 +152,7 @@ class TestMNIST(TestParallelExecutorBase):
np.mean(parallel_last_loss), single_last_loss, delta=1e-6) np.mean(parallel_last_loss), single_last_loss, delta=1e-6)
def test_simple_fc_parallel_accuracy(self): def test_simple_fc_parallel_accuracy(self):
if core.is_compiled_with_cuda(): self.check_simple_fc_parallel_accuracy(True)
self.check_simple_fc_parallel_accuracy(True)
self.check_simple_fc_parallel_accuracy(False) self.check_simple_fc_parallel_accuracy(False)
def check_batchnorm_fc_convergence(self, use_cuda, use_fast_executor): def check_batchnorm_fc_convergence(self, use_cuda, use_fast_executor):
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册