110 lines
2.7 KiB
JSON
110 lines
2.7 KiB
JSON
{
|
|
"data_path": "/root/autodl-tmp/mask-ddpm/dataset/hai/hai-21.03/train1.csv.gz",
|
|
"data_glob": "/root/autodl-tmp/mask-ddpm/mask-ddpm/example/../../dataset/hai/hai-21.03/train*.csv.gz",
|
|
"split_path": "/root/autodl-tmp/mask-ddpm/mask-ddpm/example/feature_split.json",
|
|
"stats_path": "/root/autodl-tmp/mask-ddpm/mask-ddpm/example/results/cont_stats.json",
|
|
"vocab_path": "/root/autodl-tmp/mask-ddpm/mask-ddpm/example/results/disc_vocab.json",
|
|
"out_dir": "/root/autodl-tmp/mask-ddpm/mask-ddpm/example/results",
|
|
"device": "cuda",
|
|
"timesteps": 600,
|
|
"batch_size": 12,
|
|
"seq_len": 96,
|
|
"epochs": 10,
|
|
"max_batches": 4000,
|
|
"lambda": 0.7,
|
|
"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": 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,
|
|
"cont_target": "x0",
|
|
"cont_clamp_x0": 5.0,
|
|
"use_temporal_stage1": true,
|
|
"temporal_backbone": "transformer",
|
|
"temporal_hidden_dim": 256,
|
|
"temporal_num_layers": 1,
|
|
"temporal_dropout": 0.0,
|
|
"temporal_pos_dim": 64,
|
|
"temporal_use_pos_embed": true,
|
|
"temporal_transformer_num_layers": 2,
|
|
"temporal_transformer_nhead": 4,
|
|
"temporal_transformer_ff_dim": 256,
|
|
"temporal_transformer_dropout": 0.1,
|
|
"temporal_epochs": 3,
|
|
"temporal_lr": 0.001,
|
|
"quantile_loss_weight": 0.2,
|
|
"quantile_points": [
|
|
0.05,
|
|
0.25,
|
|
0.5,
|
|
0.75,
|
|
0.95
|
|
],
|
|
"snr_weighted_loss": true,
|
|
"snr_gamma": 1.0,
|
|
"residual_stat_weight": 0.05,
|
|
"backbone_type": "transformer",
|
|
"transformer_num_layers": 3,
|
|
"transformer_nhead": 4,
|
|
"transformer_ff_dim": 512,
|
|
"transformer_dropout": 0.1,
|
|
"use_quantile_transform": true,
|
|
"quantile_bins": 1001,
|
|
"cont_bound_mode": "none",
|
|
"cont_bound_strength": 2.0,
|
|
"cont_post_calibrate": true,
|
|
"cont_post_scale": {},
|
|
"full_stats": true,
|
|
"type1_features": [
|
|
"P1_B4002",
|
|
"P2_MSD",
|
|
"P4_HT_LD",
|
|
"P1_B2004",
|
|
"P1_B3004",
|
|
"P1_B4022",
|
|
"P1_B3005"
|
|
],
|
|
"type2_features": [
|
|
"P1_B4005"
|
|
],
|
|
"type3_features": [
|
|
"P1_PCV02Z",
|
|
"P1_PCV01Z",
|
|
"P1_PCV01D",
|
|
"P1_FCV02Z"
|
|
],
|
|
"type4_features": [
|
|
"P1_PIT02",
|
|
"P2_SIT02",
|
|
"P1_FT03"
|
|
],
|
|
"type5_features": [
|
|
"P1_FT03Z"
|
|
],
|
|
"type6_features": [
|
|
"P4_HT_PO",
|
|
"P2_24Vdc",
|
|
"P2_HILout"
|
|
],
|
|
"sample_batch_size": 4,
|
|
"sample_seq_len": 96
|
|
} |