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 ·
1013 阅读 ·
0 评论