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 20:42:10 +08:00
parent f37a8ce179
commit 382c756dfe
10 changed files with 310 additions and 55 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
example/config.json
View File

@@ -1,5 +1,6 @@
{
"data_path": "../../dataset/hai/hai-21.03/train1.csv.gz",
"data_glob": "../../dataset/hai/hai-21.03/train*.csv.gz",
"split_path": "./feature_split.json",
"stats_path": "./results/cont_stats.json",
"vocab_path": "./results/disc_vocab.json",
@@ -14,5 +15,16 @@
"lr": 0.0005,
"seed": 1337,
"log_every": 10,
"ckpt_every": 50
"ckpt_every": 50,
"ema_decay": 0.999,
"use_ema": true,
"clip_k": 5.0,
"grad_clip": 1.0,
"use_condition": true,
"condition_type": "file_id",
"cond_dim": 32,
"use_tanh_eps": true,
"eps_scale": 1.0,
"sample_batch_size": 8,
"sample_seq_len": 128
}