diff --git a/train.py b/train.py index 25a9accd3be0..211cc04fb63b 100644 --- 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'])