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

update

Browse Source
This commit is contained in:
MingzheYang
2026-01-22 21:02:56 +08:00
parent 9ba96fe77e
commit 25a3647b02
2 changed files with 9 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
example/config.json
View File

@@ -7,10 +7,10 @@
"out_dir": "./results", "out_dir": "./results",
"device": "auto", "device": "auto",
"timesteps": 400, "timesteps": 400,
"batch_size": 64, "batch_size": 256,
"seq_len": 128, "seq_len": 256,
"epochs": 5, "epochs": 12,
"max_batches": 2000, "max_batches": 8000,
"lambda": 0.5, "lambda": 0.5,
"lr": 0.0005, "lr": 0.0005,
"seed": 1337, "seed": 1337,
@@ -26,5 +26,5 @@
"use_tanh_eps": true, "use_tanh_eps": true,
"eps_scale": 1.0, "eps_scale": 1.0,
"sample_batch_size": 8, "sample_batch_size": 8,
"sample_seq_len": 128 "sample_seq_len": 256
} }

4
example/platform_utils.py
View File

@@ -189,6 +189,10 @@ def resolve_path(base: Union[str, Path], target: Union[str, Path]) -> Path:
target_path = Path(target) if isinstance(target, str) else target target_path = Path(target) if isinstance(target, str) else target
if target_path.is_absolute(): if target_path.is_absolute():
return target_path return target_path
# Avoid resolving glob patterns on Windows (or any OS)
target_str = str(target_path)
if any(ch in target_str for ch in ["*", "?", "["]):
return base_path / target_path
return (base_path / target_path).resolve() return (base_path / target_path).resolve()