From 2229db523ba6391e7d4b0831bee68fa5f38a3584 Mon Sep 17 00:00:00 2001 From: tangwei12 Date: Fri, 22 Jun 2018 13:44:26 +0800 Subject: [PATCH] pserver_id init value to None --- python/paddle/fluid/trainer.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/python/paddle/fluid/trainer.py b/python/paddle/fluid/trainer.py index ad3872ab0..f191ef7df 100644 --- a/python/paddle/fluid/trainer.py +++ b/python/paddle/fluid/trainer.py @@ -131,7 +131,7 @@ class CheckpointConfig(object): self.epoch_id = 0 self.step_id = 0 self.load_serial = None - self.pserver_id = -1, + self.pserver_id = None self.lookup_table_name = None @@ -283,7 +283,7 @@ class Trainer(object): self.checkpoint_cfg.load_serial, 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( self.checkpoint_cfg.checkpoint_dir, self.checkpoint_cfg.load_serial, self.trainer_id, -- GitLab