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