-
Notifications
You must be signed in to change notification settings - Fork 12
/
params.json
executable file
·54 lines (52 loc) · 1.03 KB
/
params.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
{"params_i2t":{
"pathToData" : "../../../coco",
"vocab_path" : "./vocab_coco_full.pkl",
"vocab_threshold" : 4,
"noise_im": 64,
"noise_txt":32,
"embed_size": 512,
"max_length": 40,
"num_encoder_tokens":20,
"num_decoder_tokens": 10324,
"word_dim": 300,
"batch_size":128,
"num_gpus": 1,
"img_dim":992,
"epochs": 300,
"lambda_1": 1.0,
"lambda_2": 5000,
"lambda_3": 200.0,
"lambda_4": 0.6,
"lambda_5": 20.0,
"lambda_5_G": 25.0,
"chkpt_interval": 300,
"lr_i_e": 0.00001,
"beta_1_i_e": 0.0,
"beta_2_d": 0.999
},
"params_t2i":{
"pathToData" : "../../../coco",
"vocab_path" : "./vocab_coco_full.pkl",
"vocab_threshold" : 4,
"noise_im": 64,
"noise_txt":32,
"embed_size": 512,
"max_length": 40,
"num_encoder_tokens":20,
"num_decoder_tokens": 10324,
"word_dim": 300,
"batch_size":63,
"num_gpus": 3,
"img_dim":992,
"epochs": 300,
"lambda_1": 1.0,
"lambda_2": 2500.0,
"lambda_3": 10.0,
"lambda_4": 0.6,
"lambda_5": 1.0,
"lambda_5_G": 2000.0,
"chkpt_interval":300,
"lr_i_e": 0.0001,
"beta_1_i_e": 0.5,
"beta_2_d": 0.9
}}