Fix: evolve with wandb (ultralytics#2634)

fhkiel-mlaip · Mar 28, 2021 · 431c901 · 431c901
1 parent 59f68f8
commit 431c901
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/train.py b/train.py
@@ -439,7 +439,7 @@ def train(hyp, opt, device, tb_writer=None):
                 strip_optimizer(f)  # strip optimizers
         if opt.bucket:
             os.system(f'gsutil cp {final} gs://{opt.bucket}/weights')  # upload
-        if wandb_logger.wandb:  # Log the stripped model
+        if wandb_logger.wandb and not opt.evolve:  # Log the stripped model
             wandb_logger.wandb.log_artifact(str(final), type='model',
                                             name='run_' + wandb_logger.wandb_run.id + '_model',
                                             aliases=['last', 'best', 'stripped'])