From c021dc04181e93c6bd1f23dce690b25274a0e45f Mon Sep 17 00:00:00 2001 From: ayush chaurasia Date: Tue, 28 Dec 2021 18:38:02 +0530 Subject: [PATCH 1/3] log best.pt metrics at train end --- utils/loggers/__init__.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/utils/loggers/__init__.py b/utils/loggers/__init__.py index 7a1df2a45ea7..233d6d655402 100644 --- a/utils/loggers/__init__.py +++ b/utils/loggers/__init__.py @@ -147,9 +147,11 @@ 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: + x = {k: v for k, v in zip(self.keys[3:10], results)} # dict + self.wandb.log(x) 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: + if not self.opt.evolve: wandb.log_artifact(str(best if best.exists() else last), type='model', name='run_' + self.wandb.wandb_run.id + '_model', aliases=['latest', 'best', 'stripped']) From 3a552ff3f92ae515e2c9a4cd464bbe704b497a81 Mon Sep 17 00:00:00 2001 From: ayush chaurasia Date: Tue, 28 Dec 2021 18:43:21 +0530 Subject: [PATCH 2/3] update --- utils/loggers/__init__.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/utils/loggers/__init__.py b/utils/loggers/__init__.py index 233d6d655402..3c95dcaeb07c 100644 --- a/utils/loggers/__init__.py +++ b/utils/loggers/__init__.py @@ -151,7 +151,7 @@ def on_train_end(self, last, best, plots, epoch, results): self.wandb.log(x) 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: + if not self.opt.evolve: wandb.log_artifact(str(best if best.exists() else last), type='model', name='run_' + self.wandb.wandb_run.id + '_model', aliases=['latest', 'best', 'stripped']) From 83ae5da1d70c86f7b41c4b680cad35d2d717d864 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Thu, 30 Dec 2021 11:42:25 -0800 Subject: [PATCH 3/3] Update __init__.py --- utils/loggers/__init__.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/utils/loggers/__init__.py b/utils/loggers/__init__.py index 3c95dcaeb07c..8af5c402d5ee 100644 --- a/utils/loggers/__init__.py +++ b/utils/loggers/__init__.py @@ -147,8 +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: - x = {k: v for k, v in zip(self.keys[3:10], results)} # dict - self.wandb.log(x) + 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: