yjh0410 hace 1 año
padre
commit
e2556cc343
Se han modificado 1 ficheros con 1 adiciones y 0 borrados
  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.best_map = cfg.best_map
+        print("Best mAP metric: {}".format(self.best_map))
 
     def train(self, model):
         for epoch in range(self.start_epoch, self.cfg.max_epoch):