W&B: Allow changed in config variable ultralytics#3588

fhkiel-mlaip · Jun 11, 2021 · d009200 · d009200
1 parent 5f4b989
commit d009200
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/utils/wandb_logging/wandb_utils.py b/utils/wandb_logging/wandb_utils.py
@@ -103,7 +103,11 @@ def __init__(self, opt, name, run_id, data_dict, job_type='Training'):
                 model_artifact_name = WANDB_ARTIFACT_PREFIX + model_artifact_name
                 assert wandb, 'install wandb to resume wandb runs'
                 # Resume wandb-artifact:// runs here| workaround for not overwriting wandb.config
-                self.wandb_run = wandb.init(id=run_id, project=project, entity=entity, resume='allow')
+                self.wandb_run = wandb.init(id=run_id,
+                                            project=project,
+                                            entity=entity,
+                                            resume='allow',
+                                            allow_val_change=True)
                 opt.resume = model_artifact_name
         elif self.wandb:
             self.wandb_run = wandb.init(config=opt,
@@ -112,7 +116,8 @@ def __init__(self, opt, name, run_id, data_dict, job_type='Training'):
                                         entity=opt.entity,
                                         name=name,
                                         job_type=job_type,
-                                        id=run_id) if not wandb.run else wandb.run
+                                        id=run_id,
+                                        allow_val_change=True) if not wandb.run else wandb.run
         if self.wandb_run:
             if self.job_type == 'Training':
                 if not opt.resume: