Merge pull request #45 from rickymf4/master

modify lstm ouput for aggregated model
google · Apr 15, 2017 · 6d0c91d · 6d0c91d
2 parents e62788a + 21cc44e
commit 6d0c91d
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/frame_level_models.py b/frame_level_models.py
@@ -217,9 +217,9 @@ def create_model(self, model_input, vocab_size, num_frames, **unused_params):
     stacked_lstm = tf.contrib.rnn.MultiRNNCell(
             [
                 tf.contrib.rnn.BasicLSTMCell(
-                    lstm_size, forget_bias=1.0, state_is_tuple=False)
+                    lstm_size, forget_bias=1.0)
                 for _ in range(number_of_layers)
-                ], state_is_tuple=False)
+                ])
 
     loss = 0.0
 
@@ -231,6 +231,6 @@ def create_model(self, model_input, vocab_size, num_frames, **unused_params):
                                FLAGS.video_level_classifier_model)
 
     return aggregated_model().create_model(
-        model_input=state,
+        model_input=state[-1].h,
         vocab_size=vocab_size,
         **unused_params)