add ckpt attr to pserver python config

b81671ec · tangwei12 · 77c6b71e · b81671ec
显示空白变更内容
内联并排

Showing with 2 addition and 1 deletion

python/paddle/fluid/transpiler/distribute_transpiler.py python/paddle/fluid/transpiler/distribute_transpiler.py +2 -1

未找到文件。
--- a/python/paddle/fluid/transpiler/distribute_transpiler.py
+++ b/python/paddle/fluid/transpiler/distribute_transpiler.py
@@ -481,7 +481,8 @@ class DistributeTranspiler:
                "Fanin": self.trainer_num,
                "PrefetchBlock": prefetch_block,
                "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()