From d0092005eab3d5b3d1d0e6557674fbfbc0423378 Mon Sep 17 00:00:00 2001 From: Ayush Chaurasia Date: Sat, 12 Jun 2021 02:48:57 +0530 Subject: [PATCH] W&B: Allow changed in config variable #3588 --- utils/wandb_logging/wandb_utils.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/utils/wandb_logging/wandb_utils.py b/utils/wandb_logging/wandb_utils.py index 57ce9035a777..9975af63d02c 100644 --- 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: