簡體   English   中英

tf.train.Saver()。restore()和tf.saved_model.loader之間有什么區別

[英]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.

 
粵ICP備18138465號  © 2020-2024 STACKOOM.COM