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 <[email protected]>
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:
|