提交 7d9c9a01 编写于 作者: Y Yancey1989

update by comment

上级 f28f7faa
...@@ -15,6 +15,7 @@ ...@@ -15,6 +15,7 @@
import numpy as np import numpy as np
import argparse import argparse
import time import time
import math
import paddle import paddle
import paddle.fluid as fluid import paddle.fluid as fluid
...@@ -145,7 +146,7 @@ class TestDistMnist(unittest.TestCase): ...@@ -145,7 +146,7 @@ class TestDistMnist(unittest.TestCase):
retry_times -= 1 retry_times -= 1
def stop_pserver(self, pid): def stop_pserver(self, pid):
os.kill(pid, signal.SIGKILL) os.kill(pid, signal.SIGTERM)
def test_with_place(self): def test_with_place(self):
p = fluid.CUDAPlace(0) if core.is_compiled_with_cuda( p = fluid.CUDAPlace(0) if core.is_compiled_with_cuda(
...@@ -194,7 +195,7 @@ class TestDistMnist(unittest.TestCase): ...@@ -194,7 +195,7 @@ class TestDistMnist(unittest.TestCase):
acc_val = np.array(acc_set).mean() acc_val = np.array(acc_set).mean()
avg_loss_val = np.array(avg_loss_set).mean() avg_loss_val = np.array(avg_loss_set).mean()
if float(acc_val if float(acc_val
) > 0.2: # Smaller value to increase CI speed ) > 0.8: # Smaller value to increase CI speed
return return
else: else:
print( print(
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册