fix bug: init message manager when test phase

lib/main.py

@@ -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)
 

lib/utils/msg_manager.py

@@ -23,7 +23,9 @@ class MessageManager:
         mkdir(osp.join(save_path, "summary/"))
         self.writer = SummaryWriter(
             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
         self.logger = logging.getLogger('opengait')
         self.logger.setLevel(logging.INFO)