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

• Z时代
• 2024-01-10
• 分类：技术分享

我同时需要加载两个模型，其中一个模型的输入值依赖另外一个模型的输出值。
在 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 对象的问题。

以上是 TensorFlow 2.0 中如何同时加载多个已保存的模型？ 的全部内容， 来源链接： utcz.com/a/158248.html