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

Tune config for 32GB GPU memory

Browse Source
This commit is contained in:
MingzheYang
2026-01-29 00:01:34 +08:00
parent 24830ed164
commit 83c8d7ee01
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
example/config.json
View File

@@ -7,7 +7,7 @@
"out_dir": "./results",
"device": "auto",
"timesteps": 600,
"batch_size": 32,
"batch_size": 24,
"seq_len": 128,
"epochs": 10,
"max_batches": 4000,
@@ -26,16 +26,16 @@
"use_tanh_eps": false,
"eps_scale": 1.0,
"model_time_dim": 128,
"model_hidden_dim": 768,
"model_hidden_dim": 640,
"model_num_layers": 2,
"model_dropout": 0.1,
"model_ff_mult": 2,
"model_pos_dim": 64,
"model_use_pos_embed": true,
"backbone_type": "transformer",
"transformer_num_layers": 4,
"transformer_num_layers": 3,
"transformer_nhead": 4,
"transformer_ff_dim": 1024,
"transformer_ff_dim": 768,
"transformer_dropout": 0.1,
"disc_mask_scale": 0.9,
"cont_loss_weighting": "inv_std",