1.问题描述
第一次使用tf.layers.batch_normalization(还有tf.nn.batch_normalization和tf.contrib.layers.batch_norm),在网络(网络中使用batch_normalization的代码如下)训练完后把模型载入进行测试时显示“attempting to use unitialized value //moving_mean"什么的,在网上找了一圈之后发现都是讲batch_normalization的原理,很少针对测试时batch_normalizaiton的参数无法载入的问题。
batch_norm_params={
'training':is_training,
'momentum':0.997
}
with slim.arg_scope([slim.conv2d