From 01e1be86686060375f3f631c0d1c769be9d05a52 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Mon, 18 Oct 2021 14:25:57 +0200 Subject: [PATCH] Update __init__.py --- 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 b698c3d2db45..0b457df63c93 100644 --- a/utils/loggers/__init__.py +++ b/utils/loggers/__init__.py @@ -131,7 +131,7 @@ def on_model_save(self, last, epoch, final_epoch, best_fitness, fi): if ((epoch + 1) % self.opt.save_period == 0 and not final_epoch) and self.opt.save_period != -1: self.wandb.log_model(last.parent, self.opt, epoch, fi, best_model=best_fitness == fi) - def on_train_end(self, last, best, plots, epoch): + def on_train_end(self, last, best, plots, epoch, results): # Callback runs on training end if plots: plot_results(file=self.save_dir / 'results.csv') # save results.png