-
Notifications
You must be signed in to change notification settings - Fork 11
/
Copy pathconfig.json
58 lines (44 loc) · 1 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
{
"model": {
"name" : "VQA historical baseline (small) with classic attention",
"image":
{
"image_input": "raw",
"dim": [224, 224, 3],
"normalize": true,
"resnet_version": 50,
"finetune" : [],
"attention" : {
"mode": "classic",
"no_attention_mlp": 256
},
"cbn": {
"use_cbn": true,
"cbn_embedding_size": 512,
"excluded_scope_names": ["*"]
}
},
"glove": true,
"word_embedding_dim": 300,
"no_hidden_LSTM": 1024,
"no_LSTM_cell": 1,
"no_question_mlp": 1024,
"no_image_mlp": 1024,
"no_hidden_final_mlp": 1024,
"dropout_keep_prob": 0.5,
"loss": "soft",
"activation": "tanh"
},
"optimizer": {
"max_iters": 100000,
"learning_rate": 3e-4,
"batch_size": 20,
"step_size": 20000,
"clip_val": 5
},
"dico_name" : "dict.json",
"glove_name" : "glove_dict.pkl",
"checkpoint_path" : "checkpoint10.pth",
"merge_dataset": false,
"seed": -1
}