-
Notifications
You must be signed in to change notification settings - Fork 6
/
Copy pathconf.json
57 lines (49 loc) · 1.55 KB
/
conf.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
{
"meta":
{
"Version" : 1.0,
"Date" :"2018-10-21",
"Author" :"Minghua Zhang",
"Mail" :"zhangmh@pku.edu.cn",
"Model" :"Mean-Max Attention Autoencoder",
"Corpus" :"Toronto Books Corpus"
},
"path":
{
"train" : "%s/data/bookcorpus/train.txt",
"valid" : "%s/data/bookcorpus/valid.txt",
"decode" : "%s/data/bookcorpus/test.txt",
"vocab" : "%s/data/bookcorpus/books.vocab",
"w2v" : "%s/data/glove.840B.300d.txt",
"tasks" : "%s/data/senteval_data/",
"models" : "%s/models/"
},
"option":
{
"is_training" : false,
"model" : "Mean-Max-AAE",
"word_freq" : 1000,
"vocab_size": 21583,
"maxlen" : 32,
"minlen" : 4,
"position" : "emb",
"dim_word" : 300,
"dim_model" : 2048,
"head" : 8,
"dim_inner" : 4096,
"layer_n" : 1,
"clip_grad" : 5.0,
"drop_rate" : 0.5,
"lrate" : 0.0002,
"optimizer" : "adam",
"batch_size": 64,
"decode_bs" : 96,
"encode_bs" : 96,
"dispFreq" : 100,
"summaryFreq": 100,
"validFreq" : 500,
"decodeFreq" : 10000,
"saveFreq" : 1000,
"finish" : 100000
}
}