gpt4 book ai didi

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

转载 作者:太空宇宙 更新时间:2023-11-03 12:02:20 31 4
gpt4 key购买 nike

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

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

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

最佳答案

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

关于python - tf.train.Saver().restore() 和 tf.saved_model.loader 之间有什么区别,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/45075568/

31 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com