tf.layers.batch_normalization使用中遇到的坑
tf.layers.batch_normalization()的坑
特指TensorFlow 1
简单的使用方法:直接使用tf.layers.batch_normalization(input, is_training)
input: 需要进行BN的输入(一般在激活前使用BN)
is_training: 一般是在训练阶段设置为True,测试阶段设置为False。
坑:
在使用了batch_normalization后,需要添加代码:
update_ops = tf.get_collection(tf.Gr
原创
2020-09-28 09:50:38 ·
1049 阅读 ·
0 评论