-
Notifications
You must be signed in to change notification settings - Fork 4
/
base_phr_ite_seq.pbtxt
114 lines (114 loc) · 2.72 KB
/
base_phr_ite_seq.pbtxt
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
112
113
114
train_reader {
caption_graph_reader {
input_pattern: "vg-gt-cap/tfrecords/vg-gt-graph-zareian/train.tfrecord-00000-of-00001"
batch_size: 32
num_parallel_calls: 5
shuffle_buffer_size: 10000
prefetch_buffer_size: 1000
feature_dimensions: 1536
max_n_proposal: 20
}
}
eval_reader {
caption_graph_reader {
input_pattern: "vg-gt-cap/tfrecords/vg-gt-graph-zareian/val.tfrecord-00000-of-00001"
batch_size: 1
num_parallel_calls: 5
prefetch_buffer_size: 200
feature_dimensions: 1536
max_n_proposal: 20
}
}
test_reader {
caption_graph_reader {
input_pattern: "vg-gt-cap/tfrecords/vg-gt-graph-zareian/test.tfrecord-00000-of-00001"
batch_size: 1
num_parallel_calls: 5
prefetch_buffer_size: 200
feature_dimensions: 1536
max_n_proposal: 20
}
}
model {
[Cap2SG.ext] {
weight_decay: 1e-6
dropout_keep_prob: 0.5
parse_attribute_in_training: false
parse_attribute_in_evaluation: false
preprocess_options {
vocabulary_file: "data/vocab-zareian.txt"
closed_vocabulary_file: "data/vocab-zareian.txt"
minimum_frequency: 10
embedding_dims: 300
embedding_trainable: false
glove_vocabulary_file: "zoo/glove_word_tokens.txt"
glove_embedding_file: "zoo/glove_word_vectors.npy"
bias_mode: BIAS_MODE_ZERO
}
linguistic_options {
graph_network {
message_passing {
n_layer: 1
add_bi_directional_edges: true
}
}
}
grounding_options {
loss_weight: 1.0
two_branches: true
midn_attention_dropout_keep_prob: 1.0
}
detection_options {
loss_weight: 0.5
grounding_iou_threshold: 0.5
detection_feature_type: DET_FEATURE_DEFAULT
num_iterations: 4
predict_attributes: false
post_process {
score_thresh: 0.01
iou_thresh: 0.4
max_size_per_class: 4
max_total_size: 100
}
}
relation_options {
loss_weight: 0.5
grounding_iou_threshold: 0.5
relation_max_total_size: 100
relation_max_size_per_class: 5
relation_threshold: 0.01
}
common_sense_options {
loss_weight: 0.5
triple_loss_weight: 0.0
rnn_hidden_units: 100
rnn_input_keep_prob: 0.8
rnn_output_keep_prob: 0.8
rnn_state_keep_prob: 0.8
}
}
}
train_config {
learning_rate_schedule {
exponential_decay {
initial_learning_rate: 0.00001
decay_steps: 10000
decay_rate: 1.0
staircase: false
}
}
optimizer {
adam {
}
}
max_steps: 200000
log_step_count_steps: 100
save_summary_steps: 2000
save_checkpoints_steps: 5000
keep_checkpoint_max: 5
}
eval_config {
steps: 1000
start_delay_secs: 300
throttle_secs: 10
}