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

Reduce batch size and seq length for CUDA memory

Browse Source
This commit is contained in:
MingzheYang
2026-01-28 01:14:20 +08:00
parent 379c411153
commit f7034c3917
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
example/config.json
View File

@@ -7,8 +7,8 @@
"out_dir": "./results", "out_dir": "./results",
"device": "auto", "device": "auto",
"timesteps": 600, "timesteps": 600,
"batch_size": 32, "batch_size": 16,
"seq_len": 128, "seq_len": 96,
"epochs": 10, "epochs": 10,
"max_batches": 4000, "max_batches": 4000,
"lambda": 0.7, "lambda": 0.7,