yjh0410 1 年之前
父节点
当前提交
e2556cc343
共有 1 个文件被更改,包括 1 次插入0 次删除
  1. 1 0
      yolo/engine.py

+ 1 - 0
yolo/engine.py

@@ -73,6 +73,7 @@ class YoloTrainer(object):
         self.lr_scheduler = build_lr_scheduler(cfg, self.optimizer, args.resume)
         self.lr_scheduler = build_lr_scheduler(cfg, self.optimizer, args.resume)
 
 
         self.best_map = cfg.best_map
         self.best_map = cfg.best_map
+        print("Best mAP metric: {}".format(self.best_map))
 
 
     def train(self, model):
     def train(self, model):
         for epoch in range(self.start_epoch, self.cfg.max_epoch):
         for epoch in range(self.start_epoch, self.cfg.max_epoch):