TF 中保存恢復模型時,關於變量的name問題

在使用 tf.train.Saver() 來保存模型和使用 已有模型 來測試的時候,關於保存的變量(tf.Variable)需要注意的幾種情況:

Case1: 訓練時,變量沒有name屬性, 恢復時也沒有name屬性.
這是必須要保證 在train的代碼裏定義的變量的順序和test的變量的定義的順序 一致! 否則會報錯

Case2: 訓練時,變量有name屬性, 恢復時沒有name屬性, 或者訓練沒有name, 恢復有name.
無論如何都會報錯。

Case3(推薦): 訓練時,變量有name屬性, 恢復時有name屬性.
強烈推薦這種方式,只要name一致, 無論定義的順序如何,都不會報錯。

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章