Compare commits
1 Commits
4cd5085b33
...
feat/pusht
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
36fbf2a6b7 |
@@ -11,8 +11,10 @@ policy:
|
|||||||
logging:
|
logging:
|
||||||
backend: swanlab
|
backend: swanlab
|
||||||
mode: online
|
mode: online
|
||||||
|
name: ${exp_name}
|
||||||
|
resume: false
|
||||||
tags: ["${name}", "${task_name}", "${exp_name}", "swanlab"]
|
tags: ["${name}", "${task_name}", "${exp_name}", "swanlab"]
|
||||||
id: ${now:%Y%m%d%H%M%S}_${name}_${task_name}
|
id: null
|
||||||
group: ${exp_name}
|
group: ${exp_name}
|
||||||
|
|
||||||
dataloader:
|
dataloader:
|
||||||
|
|||||||
@@ -13,8 +13,10 @@ policy:
|
|||||||
logging:
|
logging:
|
||||||
backend: swanlab
|
backend: swanlab
|
||||||
mode: online
|
mode: online
|
||||||
|
name: ${exp_name}
|
||||||
|
resume: false
|
||||||
tags: ["${name}", "${task_name}", "${exp_name}", "swanlab"]
|
tags: ["${name}", "${task_name}", "${exp_name}", "swanlab"]
|
||||||
id: ${now:%Y%m%d%H%M%S}_${name}_${task_name}
|
id: null
|
||||||
group: ${exp_name}
|
group: ${exp_name}
|
||||||
|
|
||||||
dataloader:
|
dataloader:
|
||||||
|
|||||||
32
tests/test_pusht_swanlab_config.py
Normal file
32
tests/test_pusht_swanlab_config.py
Normal 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
|
||||||
Reference in New Issue
Block a user