-
Notifications
You must be signed in to change notification settings - Fork 3
/
config.ini
111 lines (104 loc) · 2.28 KB
/
config.ini
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
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
[EVAL]
Encoder_path = models/E2.ckpt
Decoder_path = models/D2.ckpt
vocab_path = data/vocab.pkl
embed_size = 256
hidden_size = 512
num_layers = 1
crop_size = 224
image_dir = data/val_resized2014
caption_path = data/annotations/captions_val2014.json
num_workers = 2
batch_size = 1
[EVAL2]
Encoder_path = models/v3_encoder-5-1000.ckpt
Decoder_path = models/v3_decoder-5-1000.ckpt
vocab_path = data/vocab.pkl
embed_size = 300
hidden_size = 50
num_layers = 3
crop_size = 224
image_dir = data/val_resized2014
caption_path = data/annotations/captions_val2014.json
num_workers = 2
batch_size = 1
[EVAL3]
Encoder_path = models/v3_encoder-1-3000.ckpt
Decoder_path = models/v3_decoder-1-3000.ckpt
vocab_path = data/vocab.pkl
embed_size = 512
hidden_size = 512
num_layers = 1
crop_size = 299
image_dir = data/val_resized2014
caption_path = data/annotations/captions_val2014.json
num_workers = 2
batch_size = 1
[TRAIN]
Encoder_path = None
Decoder_path = None
model_path = models/train
crop_size = 224
vocab_path = data/vocab.pkl
image_dir = data/resized2014
caption_path = data/annotations/captions_train2014.json
log_step = 10
save_step = 1000
embed_size = 256
hidden_size = 512
num_layers = 1
num_epochs = 5
batch_size = 128
num_workers = 2
learning_rate = 0.0001
[TRAIN2]
model_path = models/train2
Encoder_path = None
Decoder_path = None
crop_size = 224
vocab_path = data/vocab.pkl
image_dir = data/resized2014
caption_path = data/annotations/captions_train2014.json
log_step = 10
save_step = 1000
embed_size = 300
hidden_size = 50
num_layers = 3
num_epochs = 5
batch_size = 250
num_workers = 2
learning_rate = 0.0001
[TRAIN3]
model_path = models/train3
Encoder_path = None
Decoder_path = None
crop_size = 299
vocab_path = data/vocab.pkl
image_dir = data/resized2014
caption_path = data/annotations/captions_train2014.json
log_step = 10
save_step = 1000
embed_size = 512
hidden_size = 512
num_layers = 1
num_epochs = 5
batch_size = 128
num_workers = 2
learning_rate = 0.0001
[TRAIN4]
model_path = models/train4/
Encoder_path = None
Decoder_path = None
crop_size = 299
vocab_path = data/vocab.pkl
image_dir = data/resized2014
caption_path = data/annotations/captions_train2014.json
log_step = 10
save_step = 1000
embed_size = 512
hidden_size = 512
num_layers = 1
num_epochs = 5
batch_size = 128
num_workers = 2
learning_rate = 0.0001