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日,匈牙利詩人斐多菲寫下詩句:生命誠可貴,愛情價更高,若為自由故,兩者皆可拋...
    蛋撻的成長日記閱讀 285評論 0 3
  • 寒日換得玉宇清,碧藍靜璧澄瑩晶。 傍夜歸,略喜驚,遙望蒼穹月映星。 注:月明星稀,金星最亮。
    明哥明說閱讀 161評論 0 0
  • 交易、饅頭、平等、恐怖,字數不超過250字 遙遠的銀河系,有個名叫饅頭的星球,居民們有白面饅頭、小饅頭、花卷饅頭、...
    MioH閱讀 232評論 0 0
  • 此文的微信小程序為筆者個人發現、體驗并推薦,含主觀感情,所以僅供參考,實際效果以親自體驗為準。筆者選擇壁紙小程序考...
    By_syk閱讀 1,277評論 0 3
  • // 清除緩存的方法 - (void)deleteAll { DataBaseHandler *db = [Dat...
    iOS小菜鳥閱讀 335評論 0 1