W&B: Log best results after training ends (#6120)

* log best.pt metrics at train end * update * Update __init__.py Co-authored-by: Glenn Jocher <[email protected]>
ultralytics · Dec 30, 2021 · 9155eb8 · 9155eb8
1 parent db6ec66
commit 9155eb8
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/utils/loggers/__init__.py b/utils/loggers/__init__.py
@@ -147,6 +147,7 @@ def on_train_end(self, last, best, plots, epoch, results):
                 self.tb.add_image(f.stem, cv2.imread(str(f))[..., ::-1], epoch, dataformats='HWC')
 
         if self.wandb:
+            self.wandb.log({k: v for k, v in zip(self.keys[3:10], results)})  # log best.pt val results
             self.wandb.log({"Results": [wandb.Image(str(f), caption=f.name) for f in files]})
             # Calling wandb.log. TODO: Refactor this into WandbLogger.log_model
             if not self.opt.evolve: