Skip to content

S
seminar-breakout
Commit b6b1adae authored by Yuxin Wu's avatar Yuxin Wu
Browse files
Options

TF0.11 doesn't have fused_batch_norm

parent 4f0e1bd5
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
tensorpack/models/batch_norm.py
View file @ b6b1adae
...@@ -178,7 +178,7 @@ def BatchNormV2(x, use_local_stat=None, decay=0.9, epsilon=1e-5): ...@@ -178,7 +178,7 @@ def BatchNormV2(x, use_local_stat=None, decay=0.9, epsilon=1e-5):
else: else:
return tf.identity(xn, name='output') return tf.identity(xn, name='output')
if get_tf_version() >= 11: if get_tf_version() >= 12:
BatchNorm = BatchNormV2 BatchNorm = BatchNormV2
else: else:
logger.warn("BatchNorm might be faster if you update TensorFlow") 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