提交 a158bd91 编写于 作者: Y Yancey1989

fix ci

上级 e896926b
......@@ -27,6 +27,7 @@ import signal
SEED = 1
DTYPE = "float32"
paddle.dataset.mnist.fetch()
# random seed must set before configuring the network.
......@@ -147,7 +148,7 @@ class TestDistMnist(unittest.TestCase):
os.kill(pid, signal.SIGTERM)
def test_with_place(self):
p = fluid.CUDAPlace() if core.is_compiled_with_cuda(
p = fluid.CUDAPlace(0) if core.is_compiled_with_cuda(
) else fluid.CPUPlace()
pserver_pid = self.start_pserver(self._ps_endpoints)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册