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

在训练期间，从检查点等处恢复时，您需要使用保护程序。如果要加载导出的模型以进行推理，则只需要使用已保存的模型。