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:
MZ YANG
2026-01-26 22:51:24 +08:00
parent f82bce716f
commit 311164c22d
5 changed files with 1654 additions and 1654 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
example/results/config_used.json
View File

@@ -33,17 +33,17 @@
"model_pos_dim": 64,
"model_use_pos_embed": true,
"disc_mask_scale": 0.9,
"shuffle_buffer": 256,
"shuffle_buffer": 1024,
"cont_loss_weighting": "inv_std",
"cont_loss_eps": 1e-06,
"cont_target": "x0",
"cont_clamp_x0": 5.0,
"use_temporal_stage1": true,
"temporal_hidden_dim": 256,
"temporal_num_layers": 1,
"temporal_hidden_dim": 512,
"temporal_num_layers": 2,
"temporal_dropout": 0.0,
"temporal_epochs": 2,
"temporal_lr": 0.001,
"temporal_epochs": 5,
"temporal_lr": 0.0005,
"sample_batch_size": 8,
"sample_seq_len": 128
}