From c42e4dd6dd63a1b85e33726d6da34f76f6438f8e Mon Sep 17 00:00:00 2001
From: liuyuhui <liuyuhui@baidu.com>
Date: Sat, 26 Sep 2020 22:38:43 +0800
Subject: [PATCH] fix phase

---
 core/trainer.py | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/core/trainer.py b/core/trainer.py
index bbba6250..0ea99041 100755
--- a/core/trainer.py
+++ b/core/trainer.py
@@ -74,7 +74,9 @@ class Trainer(object):
         phase_names = envs.get_global_env(
             "runner." + self._runner_name + ".phases", None)
 
+        print("phase_names:{}".format(phase_names))
         _config = envs.load_yaml(config)
+        print("_config:{}".format(_config["phase"]))
 
         self._context["env"] = _config
         self._context["dataset"] = _config.get("dataset")
@@ -86,8 +88,11 @@ class Trainer(object):
             for phase in _config.get("phase"):
                 if phase["name"] in phase_names:
                     phases.append(phase)
-
         self._context["phases"] = phases
+        _config["phase"] = phases
+        self._context["env"] = _config
+        self._context["dataset"] = _config.get("dataset")
+        print("self._context[\"phases\"]:{}".format(self._context["phases"]))
         print("PaddleRec: Runner {} Begin".format(self._runner_name))
         self.which_engine()
         self.which_device()
@@ -254,6 +259,7 @@ class Trainer(object):
             try:
                 self.reload_train_context()
                 self.context_process(self._context)
+                print(self._context["env"]["phase"][0])
                 if self._context['is_exit']:
                     break
             except Exception as err:
-- 
GitLab