1 Commits

Author SHA1 Message Date
Logic
36fbf2a6b7 fix: avoid swanlab run collisions for pusht configs 2026-03-27 14:52:22 +08:00
3 changed files with 38 additions and 2 deletions

View File

@@ -11,8 +11,10 @@ policy:
logging:
backend: swanlab
mode: online
name: ${exp_name}
resume: false
tags: ["${name}", "${task_name}", "${exp_name}", "swanlab"]
id: ${now:%Y%m%d%H%M%S}_${name}_${task_name}
id: null
group: ${exp_name}
dataloader:

View File

@@ -13,8 +13,10 @@ policy:
logging:
backend: swanlab
mode: online
name: ${exp_name}
resume: false
tags: ["${name}", "${task_name}", "${exp_name}", "swanlab"]
id: ${now:%Y%m%d%H%M%S}_${name}_${task_name}
id: null
group: ${exp_name}
dataloader:

View File

@@ -0,0 +1,32 @@
import pathlib
from omegaconf import OmegaConf
ROOT_DIR = pathlib.Path(__file__).resolve().parents[1]
def _load_cfg(name: str):
return OmegaConf.load(ROOT_DIR / name)
def test_image_pusht_dit_swanlab_config_uses_exp_name_and_no_resume_collision():
cfg = _load_cfg('image_pusht_diffusion_policy_dit.yaml')
assert cfg.logging.backend == 'swanlab'
assert cfg.logging.mode == 'online'
assert cfg.logging.name == cfg.exp_name
assert cfg.logging.resume is False
assert cfg.logging.id is None
assert cfg.logging.group == cfg.exp_name
def test_image_pusht_dit_imf_swanlab_config_uses_exp_name_and_no_resume_collision():
cfg = _load_cfg('image_pusht_diffusion_policy_dit_imf.yaml')
assert cfg.logging.backend == 'swanlab'
assert cfg.logging.mode == 'online'
assert cfg.logging.name == cfg.exp_name
assert cfg.logging.resume is False
assert cfg.logging.id is None
assert cfg.logging.group == cfg.exp_name