ModuFlow/mask-ddpm
3
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

update2

Browse Source
This commit is contained in:
MingzheYang
2026-01-27 18:49:51 +08:00
parent 0569b66f7c
commit 8ab7c33eee
1 changed files with 8 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
example/run_all.py
View File

@@ -39,10 +39,16 @@ def main():
cfg = json.load(f)
# Resolve config path without duplicating base_dir on Windows when user passes example/config.json
if config_path.exists():
if config_path.is_absolute():
config_path = resolve_path(config_path.parent, config_path)
else:
config_path = resolve_path(base_dir, config_path)
candidate = base_dir / config_path
if candidate.exists():
config_path = resolve_path(candidate.parent, candidate)
elif config_path.exists():
config_path = resolve_path(config_path.parent, config_path)
else:
config_path = resolve_path(base_dir, config_path)
timesteps = cfg.get("timesteps", 200)
seq_len = cfg.get("sample_seq_len", cfg.get("seq_len", 64))
batch_size = cfg.get("sample_batch_size", cfg.get("batch_size", 2))