Update tf_cnn_basic.py

zrl4836 · web-flow · commit 10e162b33c28 · 2019-04-17T13:51:52.000+08:00
diff --git a/tf_cnn_basic.py b/tf_cnn_basic.py
@@ -2,7 +2,7 @@
 
 
 def BN(data, bn_momentum=0.9, name=None):
-    return tf.contrib.layers.batch_norm(data, momentum=bn_momentum, name=('%s__bn' % name))
+    return tf.layers.batch_normalization(data, momentum=bn_momentum, name=('%s__bn' % name))
 
 
 def AC(data, name=None):
