美文网首页
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