update

example/config_temporal_strong.json

@@ -0,0 +1,49 @@
+{
+  "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",
+  "out_dir": "./results",
+  "device": "auto",
+  "timesteps": 600,
+  "batch_size": 128,
+  "seq_len": 128,
+  "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,
+  "cont_loss_weighting": "inv_std",
+  "cont_loss_eps": 1e-6,
+  "cont_target": "x0",
+  "cont_clamp_x0": 5.0,
+  "shuffle_buffer": 1024,
+  "use_temporal_stage1": true,
+  "temporal_hidden_dim": 512,
+  "temporal_num_layers": 2,
+  "temporal_dropout": 0.0,
+  "temporal_epochs": 5,
+  "temporal_lr": 0.0005,
+  "sample_batch_size": 8,
+  "sample_seq_len": 128
+}