TensorFlow 2.0 中如何同时加载多个已保存的模型?

发布于 2022-09-12 01:14:17 字数 522 浏览 23 评论 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 技术交流群。

扫码二维码加入Web技术交流群

发布评论

需要 登录 才能够评论, 你可以免费 注册 一个本站的账号。
列表为空,暂无数据
我们使用 Cookies 和其他技术来定制您的体验包括您的登录状态等。通过阅读我们的 隐私政策 了解更多相关信息。 单击 接受 或继续使用网站,即表示您同意使用 Cookies 和您的相关数据。
原文