Skip to content

Commit

Permalink
Fix control_flow_ops (tflearn#380)
Browse files Browse the repository at this point in the history
Tensorflow 0.11 removes the ability to access python.control_flow_ops
tf.cond is (and was) the correct function
REF: tensorflow/tensorflow#4616
  • Loading branch information
Islandman93 authored and aymericdamien committed Oct 6, 2016
1 parent 2aa0ec8 commit 96ebbe5
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion tflearn/layers/normalization.py
Original file line number Diff line number Diff line change
Expand Up @@ -88,7 +88,7 @@ def update_mean_var():

# Retrieve variable managing training mode
is_training = tflearn.get_training_mode()
mean, var = tf.python.control_flow_ops.cond(
mean, var = tf.cond(
is_training, update_mean_var, lambda: (moving_mean, moving_variance))

try:
Expand Down

0 comments on commit 96ebbe5

Please sign in to comment.