30 lines
1.1 KiB
JSON
30 lines
1.1 KiB
JSON
{
|
|
"data_path": "F:\\Development\\modbus_diffusion\\mask-ddpm\\dataset\\hai\\hai-21.03\\train1.csv.gz",
|
|
"data_glob": "F:\\Development\\modbus_diffusion\\mask-ddpm\\mask-ddpm\\example\\..\\..\\dataset\\hai\\hai-21.03\\train*.csv.gz",
|
|
"split_path": "F:\\Development\\modbus_diffusion\\mask-ddpm\\mask-ddpm\\example\\feature_split.json",
|
|
"stats_path": "F:\\Development\\modbus_diffusion\\mask-ddpm\\mask-ddpm\\example\\results\\cont_stats.json",
|
|
"vocab_path": "F:\\Development\\modbus_diffusion\\mask-ddpm\\mask-ddpm\\example\\results\\disc_vocab.json",
|
|
"out_dir": "F:\\Development\\modbus_diffusion\\mask-ddpm\\mask-ddpm\\example\\results",
|
|
"device": "cuda",
|
|
"timesteps": 400,
|
|
"batch_size": 128,
|
|
"seq_len": 128,
|
|
"epochs": 8,
|
|
"max_batches": 3000,
|
|
"lambda": 0.5,
|
|
"lr": 0.0005,
|
|
"seed": 1337,
|
|
"log_every": 10,
|
|
"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
|
|
} |