Flush callbacks when on --evolve (ultralytics#6374)

* log best.pt metrics at train end * update * Update __init__.py * flush callbacks when using evolve Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
fhkiel-mlaip · Jan 20, 2022 · eac4f7f · eac4f7f
1 parent 42ccb47
commit eac4f7f
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 5 deletions.
diff --git a/train.py b/train.py
@@ -612,7 +612,7 @@ def main(opt, callbacks=Callbacks()):
 
             # Train mutation
             results = train(hyp.copy(), opt, device, callbacks)
-
+            callbacks = Callbacks()
             # Write mutation results
             print_mutation(results, hyp.copy(), save_dir, opt.bucket)
 

diff --git a/utils/loggers/__init__.py b/utils/loggers/__init__.py
@@ -159,10 +159,7 @@ def on_train_end(self, last, best, plots, epoch, results):
                 wandb.log_artifact(str(best if best.exists() else last), type='model',
                                    name='run_' + self.wandb.wandb_run.id + '_model',
                                    aliases=['latest', 'best', 'stripped'])
-                self.wandb.finish_run()
-            else:
-                self.wandb.finish_run()
-                self.wandb = WandbLogger(self.opt)
+            self.wandb.finish_run()
 
     def on_params_update(self, params):
         # Update hyperparams or configs of the experiment