1.11中的Tensorflow负载模型

时间:2020-06-20 18:37:36

标签: tensorflow tensorflow-estimator

我想使用TF v。1.11.0,因为它是可以在AWS Lambda上运行的最新版本(由于大小限制)。

在我使用TF 2.X之前,一切都很好。我将模型保存如下(这在TF 1.11上仍然有效)

linear_est = tf.estimator.LinearRegressor(feature_columns=feature_columns)
...
export_path = linear_est.export_saved_model("./model/", serving_input_fn)

但是我不知道如何重新加载此模型。在TF 2.X中,我只需要提供路径即可,一切都很好。现在我需要一个会话和Tag,但是我不知道应该给我什么作为Tag,我从来没有设置它。

with tf.Session(graph=tf.Graph()) as sess:
     imported = tf.saved_model.loader.load(sess, ["????"], export_dir='./model/1592675452')

然后我使用以下内容在TF 2.X中进行预测,这在TF 1.11中是否仍然有效?

def predict(example):
  return imported.signatures["predict"](examples=tf.constant([example.SerializeToString()]))

希望有人可以帮助我。

感谢与问候

0 个答案:

没有答案