Ayush Chaurasia commited on
Commit
53d4fc2
·
unverified ·
1 Parent(s): 28bff22

W&B: Allow changed in config variable #3588

Browse files
Files changed (1) hide show
  1. utils/wandb_logging/wandb_utils.py +7 -2
utils/wandb_logging/wandb_utils.py CHANGED
@@ -103,7 +103,11 @@ class WandbLogger():
103
  model_artifact_name = WANDB_ARTIFACT_PREFIX + model_artifact_name
104
  assert wandb, 'install wandb to resume wandb runs'
105
  # Resume wandb-artifact:// runs here| workaround for not overwriting wandb.config
106
- self.wandb_run = wandb.init(id=run_id, project=project, entity=entity, resume='allow')
 
 
 
 
107
  opt.resume = model_artifact_name
108
  elif self.wandb:
109
  self.wandb_run = wandb.init(config=opt,
@@ -112,7 +116,8 @@ class WandbLogger():
112
  entity=opt.entity,
113
  name=name,
114
  job_type=job_type,
115
- id=run_id) if not wandb.run else wandb.run
 
116
  if self.wandb_run:
117
  if self.job_type == 'Training':
118
  if not opt.resume:
 
103
  model_artifact_name = WANDB_ARTIFACT_PREFIX + model_artifact_name
104
  assert wandb, 'install wandb to resume wandb runs'
105
  # Resume wandb-artifact:// runs here| workaround for not overwriting wandb.config
106
+ self.wandb_run = wandb.init(id=run_id,
107
+ project=project,
108
+ entity=entity,
109
+ resume='allow',
110
+ allow_val_change=True)
111
  opt.resume = model_artifact_name
112
  elif self.wandb:
113
  self.wandb_run = wandb.init(config=opt,
 
116
  entity=opt.entity,
117
  name=name,
118
  job_type=job_type,
119
+ id=run_id,
120
+ allow_val_change=True) if not wandb.run else wandb.run
121
  if self.wandb_run:
122
  if self.job_type == 'Training':
123
  if not opt.resume: