Clean artifacts and update example pipeline
This commit is contained in:
18
example/config.json
Normal file
18
example/config.json
Normal file
@@ -0,0 +1,18 @@
|
||||
{
|
||||
"data_path": "../../dataset/hai/hai-21.03/train1.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": 400,
|
||||
"batch_size": 64,
|
||||
"seq_len": 128,
|
||||
"epochs": 5,
|
||||
"max_batches": 2000,
|
||||
"lambda": 0.5,
|
||||
"lr": 0.0005,
|
||||
"seed": 1337,
|
||||
"log_every": 10,
|
||||
"ckpt_every": 50
|
||||
}
|
||||
Reference in New Issue
Block a user