-
Notifications
You must be signed in to change notification settings - Fork 0
/
config.json
65 lines (65 loc) · 1.76 KB
/
config.json
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
{
"ae_model_type":"transformer",
"enc_hparams":{
"transformer":{
"enc_d_model": 64,
"enc_nhead": 4,
"enc_dff":128,
"enc_n_layers":6,
"enc_dropout":0.2
},
"rnn":{
"enc_hsize": 128,
"enc_n_layers": 4,
"enc_dropout": 0.05
}
},
"rel_hparams":{
"rel_d_model":64,
"rel_nhead":4,
"rel_dff":256,
"rel_n_layers":6,
"rel_dropout":0.2,
"rel_gmm_num_components":10
},
"dec_hparams":{
"transformer":{
"dec_gmm_num_components":20,
"dec_layer_features":[512,512,512,512]
},
"rnn":{
"dec_hsize": 128,
"dec_n_layers": 6,
"dec_dim_layer": 256,
"dec_dropout": 0.05,
"dec_gmm_num_components":20
}
},
"cose_model_params":{
"size_embedding":8,
"num_predictive_inputs": 32,
"end_positions": false
},
"training_params":{
"input_type": "hybrid",
"replace_padding": false,
"stop_predictive_grad": true,
"num_epochs":200,
"lr_ae":1e-3,
"lr_pos_pred":1e-3,
"lr_emb_pred":1e-3
},
"general_config":{
"use_gpu": true,
"root_path":"./",
"save_path":"weights_trained/",
"diagrams_img_path":"diagramas_img/",
"train_dataset_path":"/data/ajimenez/cose/train/",
"validation_dataset_path":"/data/ajimenez/cose/valid/",
"test_dataset_path": "/data/ajimenez/cose/test/",
"num_backups":50,
"model_path":"weights/",
"save_weights":true,
"stats_path": "/data/jcabrera/didi_wo_text/didi_wo_text-stats-origin_abs_pos.json"
}
}