提交 2229db52 编写于 作者: T tangwei12

pserver_id init value to None

上级 8e01f3b9
...@@ -131,7 +131,7 @@ class CheckpointConfig(object): ...@@ -131,7 +131,7 @@ class CheckpointConfig(object):
self.epoch_id = 0 self.epoch_id = 0
self.step_id = 0 self.step_id = 0
self.load_serial = None self.load_serial = None
self.pserver_id = -1, self.pserver_id = None
self.lookup_table_name = None self.lookup_table_name = None
...@@ -283,7 +283,7 @@ class Trainer(object): ...@@ -283,7 +283,7 @@ class Trainer(object):
self.checkpoint_cfg.load_serial, self.checkpoint_cfg.load_serial,
self.startup_program) self.startup_program)
if self.checkpoint_cfg.pserver_id != -1: if not self.checkpoint_cfg.pserver_id:
epoch_id, step_id = io.load_trainer_args( epoch_id, step_id = io.load_trainer_args(
self.checkpoint_cfg.checkpoint_dir, self.checkpoint_cfg.checkpoint_dir,
self.checkpoint_cfg.load_serial, self.trainer_id, self.checkpoint_cfg.load_serial, self.trainer_id,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册