Skip to content

Commit

Permalink
Merge pull request #3 from rajshah4/master
Browse files Browse the repository at this point in the history
Fix for Tensorflow 7.1
  • Loading branch information
hardmaru committed Mar 21, 2016
2 parents 218d8af + 9fb8367 commit 0a148b0
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions model.py
Original file line number Diff line number Diff line change
Expand Up @@ -44,10 +44,10 @@ def __init__(self, args, infer=False):
inputs = tf.split(1, args.seq_length, self.input_data)
inputs = [tf.squeeze(input_, [1]) for input_ in inputs]

outputs, states = seq2seq.rnn_decoder(inputs, self.initial_state, cell, loop_function=None, scope='rnnlm')
outputs, last_state = seq2seq.rnn_decoder(inputs, self.initial_state, cell, loop_function=None, scope='rnnlm')
output = tf.reshape(tf.concat(1, outputs), [-1, args.rnn_size])
output = tf.nn.xw_plus_b(output, output_w, output_b)
self.final_state = states[-1]
self.final_state = last_state

# reshape target data so that it is compatible with prediction shape
flat_target_data = tf.reshape(self.target_data,[-1, 3])
Expand Down

0 comments on commit 0a148b0

Please sign in to comment.