From 5c8397a88fe6b062be0c0725bbd14a2c8d4fc2e9 Mon Sep 17 00:00:00 2001 From: tangwei12 Date: Fri, 8 Jun 2018 19:49:10 +0800 Subject: [PATCH] remove chief in test --- python/paddle/fluid/tests/unittests/test_checkpoint.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/python/paddle/fluid/tests/unittests/test_checkpoint.py b/python/paddle/fluid/tests/unittests/test_checkpoint.py index cf70dfd448..e22400a045 100644 --- a/python/paddle/fluid/tests/unittests/test_checkpoint.py +++ b/python/paddle/fluid/tests/unittests/test_checkpoint.py @@ -66,9 +66,9 @@ class TestCheckpoint(unittest.TestCase): exe = fluid.Executor(self.place) for i in xrange(10): - fluid.io.save_checkpoint( - exe, config.checkpoint_dir, self.trainer_id, self.chief, - trainer_args, program, config.max_num_checkpoints) + fluid.io.save_checkpoint(exe, config.checkpoint_dir, + self.trainer_id, trainer_args, program, + config.max_num_checkpoints) if __name__ == '__main__': -- GitLab