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 对象的问题。

阅读 3.1k
撰写回答
你尚未登录,登录后可以
  • 和开发者交流问题的细节
  • 关注并接收问题和回答的更新提醒
  • 参与内容的编辑和改进,让解决方法与时俱进
推荐问题