瀏覽代碼

W&B: Allow changed in config variable #3588

modifyDataloader
Ayush Chaurasia GitHub 3 年之前
父節點
當前提交
53d4fc2e26
沒有發現已知的金鑰在資料庫的簽署中 GPG 金鑰 ID: 4AEE18F83AFDEB23
共有 1 個檔案被更改,包括 7 行新增2 行删除
  1. +7
    -2
      utils/wandb_logging/wandb_utils.py

+ 7
- 2
utils/wandb_logging/wandb_utils.py 查看文件

@@ -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:

Loading…
取消
儲存