|
|
@@ -103,7 +103,11 @@ class WandbLogger(): |
|
|
|
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 @@ class WandbLogger(): |
|
|
|
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: |