diff --git a/train.py b/train.py index eeeb549a6341..403428ca60f6 100644 --- a/train.py +++ b/train.py @@ -484,7 +484,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'])