美文网首页
slim.batch_norm无法保存参数

slim.batch_norm无法保存参数

作者: zhenggeaza | 来源:发表于2017-09-18 10:05 被阅读0次

slim.batch_norm中的moving_mean和moving_variance不是trainable的,所以使用saver = tf.train.Saver(tf.trainable_variables(), max_to_keep=3)无法保存,应该改为:

var_list = tf.trainable_variables()

g_list = tf.global_variables()

bn_moving_vars = [g for g in g_list if 'moving_mean' in g.name]

bn_moving_vars += [g for g in g_list if 'moving_variance' in g.name]

var_list += bn_moving_vars

saver = tf.train.Saver(var_list=var_list, max_to_keep=3)

相关文章

网友评论

      本文标题:slim.batch_norm无法保存参数

      本文链接:https://www.haomeiwen.com/subject/rlvvsxtx.html