Ayush Chaurasia
commited on
W&B: Allow changed in config variable #3588
Browse files
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,
|
|
|
|
|
|
|
|
|
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
|
|
|
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:
|