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'] engine_cfg = cfgs['trainer_cfg'] if training else cfgs['evaluator_cfg']
output_path = os.path.join('output/', cfgs['data_cfg']['dataset_name'], output_path = os.path.join('output/', cfgs['data_cfg']['dataset_name'],
cfgs['model_cfg']['model'], engine_cfg['save_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, if training:
engine_cfg['restore_hint'] if isinstance(engine_cfg['restore_hint'], (int)) else 0) 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) msg_mgr.log_info(engine_cfg)
lib/utils/msg_manager.py
+2
View File
@@ -23,7 +23,9 @@ class MessageManager:
mkdir(osp.join(save_path, "summary/")) mkdir(osp.join(save_path, "summary/"))
self.writer = SummaryWriter( self.writer = SummaryWriter(
osp.join(save_path, "summary/"), purge_step=self.iteration) osp.join(save_path, "summary/"), purge_step=self.iteration)
self.init_logger(save_path, log_to_file)
def init_logger(self, save_path, log_to_file):
# init logger # init logger
self.logger = logging.getLogger('opengait') self.logger = logging.getLogger('opengait')
self.logger.setLevel(logging.INFO) self.logger.setLevel(logging.INFO)