fix save step bug

369d2592 · liuyuhui · 1671a378 · 369d2592
隐藏空白更改
内联并排

Showing with 6 addition and 6 deletion

core/trainers/framework/runner.py core/trainers/framework/runner.py +6 -6

未找到文件。
--- a/core/trainers/framework/runner.py
+++ b/core/trainers/framework/runner.py
@@ -209,14 +209,14 @@ class RunnerBase(object):
                    if save_step_interval >= 1 and batch_id % save_step_interval == 0 and context[
                            "is_infer"] == False:
-                        if context["fleet_mode"]:
+                        if context["fleet_mode"].upper() == "PS":
-                            if context["fleet_mode"].upper() == "PS":
+                            train_prog = context["model"][model_dict["name"]][
-                                train_prog = context["model"][model_dict[
+                                "main_program"]
-                                    "name"]]["main_program"]
+                            print("condition 1")
-                        elif not context["is_fleet"] or context[
+                        else:
-                                "fleet_mode"].upper() == "COLLECTIVE":
                            train_prog = context["model"][model_dict["name"]][
                                "default_main_program"]
+                            print("condition 2")
                        startup_prog = context["model"][model_dict["name"]][
                            "startup_program"]
                        with fluid.program_guard(train_prog, startup_prog):