Ayush Chaurasia glenn-jocher commited on
Commit
9155eb8
1 Parent(s): db6ec66

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

Browse files

* log best.pt metrics at train end

* update

* Update __init__.py

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>

Files changed (1) hide show
  1. utils/loggers/__init__.py +1 -0
utils/loggers/__init__.py CHANGED
@@ -147,6 +147,7 @@ class Loggers():
147
  self.tb.add_image(f.stem, cv2.imread(str(f))[..., ::-1], epoch, dataformats='HWC')
148
 
149
  if self.wandb:
 
150
  self.wandb.log({"Results": [wandb.Image(str(f), caption=f.name) for f in files]})
151
  # Calling wandb.log. TODO: Refactor this into WandbLogger.log_model
152
  if not self.opt.evolve:
 
147
  self.tb.add_image(f.stem, cv2.imread(str(f))[..., ::-1], epoch, dataformats='HWC')
148
 
149
  if self.wandb:
150
+ self.wandb.log({k: v for k, v in zip(self.keys[3:10], results)}) # log best.pt val results
151
  self.wandb.log({"Results": [wandb.Image(str(f), caption=f.name) for f in files]})
152
  # Calling wandb.log. TODO: Refactor this into WandbLogger.log_model
153
  if not self.opt.evolve: