TensorFlow 2.0 中如何同时加载多个已保存的模型?
我同时需要加载两个模型,其中一个模型的输入值依赖另外一个模型的输出值。
在 TensorFlow 1.x 版本中,我们可以通过创建多个 tf.Graph
来实现这个需求。
import tensorflow as tf
g1 = tf.Graph()
g2 = tf.Graph()
with g1.as_default():
model1 = tf.saved_model.load('model1/').signatures['serving_default']
with g2.as_default():
model2 = tf.saved_model.load('model2/').signatures['serving_default']
但在 TF2.0 中虽然这个方法已经被弃用,而直接加载两个模型会导致前一个模型的资源丢失而无法使用。如果使用 tf.Graph
的方式进行加载的话,就有 tf.Tensor
无法转换成 numpy 对象的问题。
如果你对这篇内容有疑问,欢迎到本站社区发帖提问 参与讨论,获取更多帮助,或者扫码二维码加入 Web 技术交流群。
绑定邮箱获取回复消息
由于您还没有绑定你的真实邮箱,如果其他用户或者作者回复了您的评论,将不能在第一时间通知您!
发布评论