optimize python checkpint dir config

shanyi15-patch-3
tangwei12 7 years ago
parent b6ee59ae25
commit e130bf3752

@ -543,6 +543,11 @@ class DistributeTranspiler:
serial_number = self._get_lastest_checkpoint_dir(self.checkpoint_dir)
startup_prog.global_block().create_var(
name=SERIAL_VAR_NAME,
persistable=True,
type=core.VarDesc.VarType.RAW)
startup_prog.global_block().append_op(
type="checkpoint_load",
inputs={"X": load_vars},
@ -612,6 +617,11 @@ class DistributeTranspiler:
serial_number = self._get_lastest_checkpoint_dir(self.checkpoint_dir)
s_prog.global_block().create_var(
name=SERIAL_VAR_NAME,
persistable=True,
type=core.VarDesc.VarType.RAW)
s_prog.global_block().append_op(
type="checkpoint_load",
inputs={"X": load_vars},

Loading…
Cancel
Save