-
Notifications
You must be signed in to change notification settings - Fork 130
/
config_real
40 lines (39 loc) · 1.94 KB
/
config_real
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
{
"task": "train",
"device": "gpuX",
"multiprocessing": true,
"batching": "default",
"cache_size": "256G",
"max_seqs": 10,
"batch_size": 600000,
"nadam": 1,
"learning_rate": 0.0005,
"learning_rates": "{1: 0.0005, 25: 0.0003, 35: 0.0001}",
"num_epochs": 200,
"update_batch_size": 2000000,
"reinit": 0,
"log_verbosity": 5,
"train": "features/raw/train.1.h5,features/raw/train.2.h5",
"dev": "features/raw/train_valid.h5",
"start_epoch": 1,
"num_inputs": 1,
"num_outputs": {"data": [1,2], "classes": [79,1], "sizes": [2,1]},
"network": {
"classes_source": {"class": "source", "data_key": "sizes", "from": [], "n_out": 2, "dtype":"int32"},
"1Dto2D": {"class": "1Dto2D", "from" : ["data", "classes_source"] },
"conv0": { "class" : "conv2", "n_features" : 15, "filter" : [3, 3], "pool_size" : [2, 2], "from" : ["1Dto2D"] },
"mdlstm0": { "class" : "mdlstm", "n_out" : 30, "dropout": 0.25, "from" : ["conv0"] },
"conv1" : { "class" : "conv2", "n_features" : 45, "dropout": 0.25, "filter" : [3, 3], "pool_size" : [2, 2], "from" : ["mdlstm0"] },
"mdlstm1": { "class" : "mdlstm", "n_out" : 60, "dropout": 0.25, "from" : ["conv1"] },
"conv2" : { "class" : "conv2", "n_features" : 75, "dropout": 0.25, "filter" : [3, 3], "pool_size" : [2, 2], "from" : ["mdlstm1"] },
"mdlstm2": { "class" : "mdlstm", "n_out" : 90, "dropout": 0.25, "from" : ["conv2"] },
"conv3" : { "class" : "conv2", "n_features" : 105, "dropout": 0.25, "filter" : [3, 3], "pool_size" : [1, 1], "from" : ["mdlstm2"] },
"mdlstm3": { "class" : "mdlstm", "n_out" : 120, "dropout": 0.25, "from" : ["conv3"] },
"conv4" : { "class" : "conv2", "n_features" : 105, "dropout": 0.25, "filter" : [3, 3], "pool_size" : [1, 1], "from" : ["mdlstm3"] },
"mdlstm4": { "class" : "mdlstm", "n_out" : 120, "dropout": 0.25, "from" : ["conv4"] },
"output": { "class" : "softmax", "loss": "ctc", "dropout": 0.25, "from" : ["mdlstm4"] }
},
"ctc_prior_file": "priors/mdlstm_real",
"model": "models/mdlstm_real",
"log": "log/mdlstm_real"
}