tf.train.Saver().restore() 和 tf.saved_model.loader 之间的区别是什么？

Question

4

我想了解tf.train.Saver().restore()和tf.saved_model.loader()之间的区别。

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

但我不知道何时应选择restore()或loader()？

- xuan lee

1个回答

网页内容由stack overflow 提供, 点击上面的

可以查看英文原文，
原文链接

- Alexandre Passos · Answer 1

在训练过程中、从检查点恢复等情况下，您需要使用保存器。只有在加载导出的模型进行推理时，才需要使用已保存的模型。