From b81671ecf214edca344cce12da51d6f0e1d21a66 Mon Sep 17 00:00:00 2001 From: tangwei12 Date: Thu, 10 May 2018 20:03:58 +0800 Subject: [PATCH] add ckpt attr to pserver python config --- python/paddle/fluid/transpiler/distribute_transpiler.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/python/paddle/fluid/transpiler/distribute_transpiler.py b/python/paddle/fluid/transpiler/distribute_transpiler.py index 640ac9f085e..8cd7cd5d3a9 100644 --- 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() -- GitLab