Skip to content

Commit c42e4dd

Browse files
committed
fix phase
1 parent 7cfe354 commit c42e4dd

File tree

1 file changed

+7
-1
lines changed

1 file changed

+7
-1
lines changed

core/trainer.py

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -74,7 +74,9 @@ def __init__(self, config=None):
7474
phase_names = envs.get_global_env(
7575
"runner." + self._runner_name + ".phases", None)
7676

77+
print("phase_names:{}".format(phase_names))
7778
_config = envs.load_yaml(config)
79+
print("_config:{}".format(_config["phase"]))
7880

7981
self._context["env"] = _config
8082
self._context["dataset"] = _config.get("dataset")
@@ -86,8 +88,11 @@ def __init__(self, config=None):
8688
for phase in _config.get("phase"):
8789
if phase["name"] in phase_names:
8890
phases.append(phase)
89-
9091
self._context["phases"] = phases
92+
_config["phase"] = phases
93+
self._context["env"] = _config
94+
self._context["dataset"] = _config.get("dataset")
95+
print("self._context[\"phases\"]:{}".format(self._context["phases"]))
9196
print("PaddleRec: Runner {} Begin".format(self._runner_name))
9297
self.which_engine()
9398
self.which_device()
@@ -254,6 +259,7 @@ def run(self):
254259
try:
255260
self.reload_train_context()
256261
self.context_process(self._context)
262+
print(self._context["env"]["phase"][0])
257263
if self._context['is_exit']:
258264
break
259265
except Exception as err:

0 commit comments

Comments
 (0)