crosstyan/OpenGait
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix bug: init message manager when test phase

Browse Source
This commit is contained in:
梁峻豪
2021-10-20 22:55:43 +08:00
parent c8e5bc1cbe
commit b4c569e835
2 changed files with 7 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
lib/main.py
+5 -2
View File
@@ -24,8 +24,11 @@ def initialization(cfgs, training):
engine_cfg = cfgs['trainer_cfg'] if training else cfgs['evaluator_cfg']
output_path = os.path.join('output/', cfgs['data_cfg']['dataset_name'],
cfgs['model_cfg']['model'], engine_cfg['save_name'])
msg_mgr.init_manager(output_path, opt.log_to_file, engine_cfg['log_iter'] if training else 0,
engine_cfg['restore_hint'] if isinstance(engine_cfg['restore_hint'], (int)) else 0)
if training:
msg_mgr.init_manager(output_path, opt.log_to_file, engine_cfg['log_iter'],
engine_cfg['restore_hint'] if isinstance(engine_cfg['restore_hint'], (int)) else 0)
else:
msg_mgr.init_logger(output_path, opt.log_to_file)
msg_mgr.log_info(engine_cfg)