fix for shakeshake2_py with equal=True (#510)

tensorflow · Jan 12, 2018 · cc43389 · cc43389
1 parent f55462a
commit cc43389
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tensor2tensor/layers/common_layers.py b/tensor2tensor/layers/common_layers.py
@@ -76,7 +76,7 @@ def shakeshake2_py(x, y, equal=False, individual=False):
   """The shake-shake sum of 2 tensors, python version."""
   if equal:
     alpha = 0.5
-  if individual:
+  elif individual:
     alpha = tf.random_uniform(tf.get_shape(x)[:1])
   else:
     alpha = tf.random_uniform([])