连续型特征在时许相关性上的不足
This commit is contained in:
@@ -6,11 +6,11 @@
|
||||
"vocab_path": "./results/disc_vocab.json",
|
||||
"out_dir": "./results",
|
||||
"device": "auto",
|
||||
"timesteps": 400,
|
||||
"timesteps": 600,
|
||||
"batch_size": 128,
|
||||
"seq_len": 128,
|
||||
"epochs": 8,
|
||||
"max_batches": 3000,
|
||||
"epochs": 10,
|
||||
"max_batches": 4000,
|
||||
"lambda": 0.5,
|
||||
"lr": 0.0005,
|
||||
"seed": 1337,
|
||||
@@ -23,8 +23,17 @@
|
||||
"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,
|
||||
"sample_batch_size": 8,
|
||||
"sample_seq_len": 128
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user