bnlstm
bnlstm copied to clipboard
some error on tensorflow 1.8.0
1.tf.cond(training,batch_statistics, population_statistics)
in line 138.
training is python bool,but cond need tf.bool. so modify it like this:
tf.cond(tf.cast(training, tf.bool), batch_statistics, population_statistics)
2.In line 81 i, j, f, o = tf.split(1, 4, hidden)
not specify name arg. Modify like this:
i, j, f, o = tf.split(axis=1, num_or_size_splits=4, value=hidden)
Thanks for your implementation.