slim.batch_norm无法保存参数

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)

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容

  • 你觉得爱情是什么? 1849年7月31日,匈牙利诗人斐多菲写下诗句:生命诚可贵,爱情价更高,若为自由故,两者皆可抛...
    蛋挞的成长日记阅读 2,312评论 0 3
  • 寒日换得玉宇清,碧蓝静璧澄莹晶。 傍夜归,略喜惊,遥望苍穹月映星。 注:月明星稀,金星最亮。
    竹影灯阅读 1,066评论 0 0
  • 交易、馒头、平等、恐怖,字数不超过250字 遥远的银河系,有个名叫馒头的星球,居民们有白面馒头、小馒头、花卷馒头、...
    MioH阅读 1,573评论 0 0
  • 此文的微信小程序为笔者个人发现、体验并推荐,含主观感情,所以仅供参考,实际效果以亲自体验为准。笔者选择壁纸小程序考...
    By_syk阅读 5,232评论 0 3
  • // 清除缓存的方法 - (void)deleteAll { DataBaseHandler *db = [Dat...
    iOS小菜鸟阅读 2,675评论 0 1