optimize python checkpint dir config

e130bf37 · tangwei12 · b6ee59ae · e130bf37
隐藏空白更改
内联并排

Showing with 10 addition and 0 deletion

python/paddle/fluid/transpiler/distribute_transpiler.py python/paddle/fluid/transpiler/distribute_transpiler.py +10 -0

未找到文件。
--- a/python/paddle/fluid/transpiler/distribute_transpiler.py
+++ b/python/paddle/fluid/transpiler/distribute_transpiler.py
@@ -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},