Commit b6b1adae authored by Yuxin Wu's avatar Yuxin Wu

TF0.11 doesn't have fused_batch_norm

parent 4f0e1bd5
......@@ -178,7 +178,7 @@ def BatchNormV2(x, use_local_stat=None, decay=0.9, epsilon=1e-5):
else:
return tf.identity(xn, name='output')
if get_tf_version() >= 11:
if get_tf_version() >= 12:
BatchNorm = BatchNormV2
else:
logger.warn("BatchNorm might be faster if you update TensorFlow")
......
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