diff --git a/core/trainer.py b/core/trainer.py index bbba6250529283d24389e2719b7110f8aa321973..0ea9904157d5bc80f0598c4f130701c9d6c626b6 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: