提交 b81671ec 编写于 作者: T tangwei12

add ckpt attr to pserver python config

上级 77c6b71e
...@@ -481,7 +481,8 @@ class DistributeTranspiler: ...@@ -481,7 +481,8 @@ class DistributeTranspiler:
"Fanin": self.trainer_num, "Fanin": self.trainer_num,
"PrefetchBlock": prefetch_block, "PrefetchBlock": prefetch_block,
"sync_mode": self.sync_mode, "sync_mode": self.sync_mode,
"grad_to_block_id": grad_to_block_id "grad_to_block_id": grad_to_block_id,
"Checkpoint": "/tmp/tangwei_ckpt/"
}) })
pserver_program.sync_with_cpp() pserver_program.sync_with_cpp()
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册