This commit is contained in:
MZ YANG
2026-01-24 00:13:25 +08:00
parent e0949da18d
commit d953af9f2c
4 changed files with 1491 additions and 1468 deletions

View File

@@ -6,12 +6,12 @@
"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,
"timesteps": 600,
"batch_size": 128,
"seq_len": 128,
"epochs": 8,
"max_batches": 3000,
"lambda": 0.5,
"epochs": 10,
"max_batches": 4000,
"lambda": 0.7,
"lr": 0.0005,
"seed": 1337,
"log_every": 10,
@@ -23,8 +23,19 @@
"use_condition": true,
"condition_type": "file_id",
"cond_dim": 32,
"use_tanh_eps": true,
"use_tanh_eps": false,
"eps_scale": 1.0,
"model_time_dim": 128,
"model_hidden_dim": 512,
"model_num_layers": 2,
"model_dropout": 0.1,
"model_ff_mult": 2,
"model_pos_dim": 64,
"model_use_pos_embed": true,
"disc_mask_scale": 0.9,
"shuffle_buffer": 256,
"cont_loss_weighting": "inv_std",
"cont_loss_eps": 1e-06,
"sample_batch_size": 8,
"sample_seq_len": 128
}