slim.batch_norm無法保存參數(shù)

slim.batch_norm中的moving_mean和moving_variance不是trainable的,所以使用saver = tf.train.Saver(tf.trainable_variables(), max_to_keep=3)無法保存,應(yīng)該改為:

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)

最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時請結(jié)合常識與多方信息審慎甄別。
平臺聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點,簡書系信息發(fā)布平臺,僅提供信息存儲服務(wù)。

相關(guān)閱讀更多精彩內(nèi)容

  • 你覺得愛情是什么? 1849年7月31日,匈牙利詩人斐多菲寫下詩句:生命誠可貴,愛情價更高,若為自由故,兩者皆可拋...
    蛋撻的成長日記閱讀 355評論 0 3
  • 寒日換得玉宇清,碧藍(lán)靜璧澄瑩晶。 傍夜歸,略喜驚,遙望蒼穹月映星。 注:月明星稀,金星最亮。
    竹影燈閱讀 169評論 0 0
  • 交易、饅頭、平等、恐怖,字?jǐn)?shù)不超過250字 遙遠(yuǎn)的銀河系,有個名叫饅頭的星球,居民們有白面饅頭、小饅頭、花卷饅頭、...
    MioH閱讀 258評論 0 0
  • 此文的微信小程序為筆者個人發(fā)現(xiàn)、體驗并推薦,含主觀感情,所以僅供參考,實際效果以親自體驗為準(zhǔn)。筆者選擇壁紙小程序考...
    By_syk閱讀 1,491評論 0 3
  • // 清除緩存的方法 - (void)deleteAll { DataBaseHandler *db = [Dat...
    iOS小菜鳥閱讀 381評論 0 1

友情鏈接更多精彩內(nèi)容