Skip to content

Commit 6d0c91d

Browse files
authored
Merge pull request google#45 from rickymf4/master
modify lstm ouput for aggregated model
2 parents e62788a + 21cc44e commit 6d0c91d

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

frame_level_models.py

+3-3
Original file line numberDiff line numberDiff line change
@@ -217,9 +217,9 @@ def create_model(self, model_input, vocab_size, num_frames, **unused_params):
217217
stacked_lstm = tf.contrib.rnn.MultiRNNCell(
218218
[
219219
tf.contrib.rnn.BasicLSTMCell(
220-
lstm_size, forget_bias=1.0, state_is_tuple=False)
220+
lstm_size, forget_bias=1.0)
221221
for _ in range(number_of_layers)
222-
], state_is_tuple=False)
222+
])
223223

224224
loss = 0.0
225225

@@ -231,6 +231,6 @@ def create_model(self, model_input, vocab_size, num_frames, **unused_params):
231231
FLAGS.video_level_classifier_model)
232232

233233
return aggregated_model().create_model(
234-
model_input=state,
234+
model_input=state[-1].h,
235235
vocab_size=vocab_size,
236236
**unused_params)

0 commit comments

Comments
 (0)