python - tf.train.Saver().restore() 和 tf.saved_model.loader 之间有什么区别

Question

我想知道 tf.train.Saver().restore() 和 tf.saved_model.loader()。

据我所知，tf.train.Saver().restore() 从checkpoint 文件中恢复之前保存的变量；并且 tf.saved_model.loader() 从 pb 文件加载图形 def。

但我不知道什么时候应该选择 restore() 或 loader()？

Answer 1

在训练、从检查点恢复等过程中，您想使用保护程序。如果您正在加载导出的模型进行推理，您只想使用保存的模型。