我想在TensorFlow中使用批量规范化.我找到了相关的C++源代码core/ops/nn_ops.cc.但是,我没有在tensorflow.org上找到它.
BN在MLP和CNN中有不同的语义,所以我不确定这个BN究竟是做什么的.
我没有找到一个叫做的方法MovingMoments.
我正在训练以下模型:
with slim.arg_scope(inception_arg_scope(is_training=True)):
logits_v, endpoints_v = inception_v3(all_v, num_classes=25, is_training=True, dropout_keep_prob=0.8,
spatial_squeeze=True, reuse=reuse_variables, scope='vis')
logits_p, endpoints_p = inception_v3(all_p, num_classes=25, is_training=True, dropout_keep_prob=0.8,
spatial_squeeze=True, reuse=reuse_variables, scope='pol')
pol_features = endpoints_p['pol/features']
vis_features = endpoints_v['vis/features']
eps = 1e-08
loss = tf.sqrt(tf.maximum(tf.reduce_sum(tf.square(pol_features - vis_features), axis=1, keep_dims=True), eps))
# rest of code
saver = tf.train.Saver(tf.global_variables())
Run Code Online (Sandbox Code Playgroud)
哪里
def inception_arg_scope(weight_decay=0.00004,
batch_norm_decay=0.9997,
batch_norm_epsilon=0.001, is_training=True):
normalizer_params = {
'decay': batch_norm_decay,
'epsilon': batch_norm_epsilon,
'is_training': is_training
}
normalizer_fn = tf.contrib.layers.batch_norm
# Set weight_decay for weights in Conv and FC layers.
with slim.arg_scope([slim.conv2d, slim.fully_connected],
weights_regularizer=slim.l2_regularizer(weight_decay)): …Run Code Online (Sandbox Code Playgroud)