Commit 31dccbad authored by Neal Wu's avatar Neal Wu Committed by GitHub
Browse files

Merge pull request #1025 from shahparth123/patch-1

Updated concat_v2 to concat for 1.0 compatibility
parents 9baf6eac dad7dcbd
......@@ -264,7 +264,7 @@ class ShowAndTellModel(object):
if self.mode == "inference":
# In inference mode, use concatenated states for convenient feeding and
# fetching.
tf.concat_v2(initial_state, 1, name="initial_state")
tf.concat(initial_state, 1, name="initial_state")
# Placeholder for feeding a batch of concatenated states.
state_feed = tf.placeholder(dtype=tf.float32,
......@@ -278,7 +278,7 @@ class ShowAndTellModel(object):
state=state_tuple)
# Concatentate the resulting state.
tf.concat_v2(state_tuple, 1, name="state")
tf.concat(state_tuple, 1, name="state")
else:
# Run the batch of sequence embeddings through the LSTM.
sequence_length = tf.reduce_sum(self.input_mask, 1)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment