[英]what the differences between tf.train.Saver().restore() and tf.saved_model.loader
我想知道tf.train.Saver().restore()
和tf.saved_model.loader()
之間的區別。
據我所知, tf.train.Saver().restore()
從checkpoint
文件恢復以前保存的變量; 和tf.saved_model.loader()
從pb
文件加載圖形def。
但我不知道何時應該選擇restore()
或loader()
?
在訓練期間,從檢查點等處恢復時,您需要使用保護程序。 如果要加載導出的模型以進行推理,則只需要使用已保存的模型。
聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.