我想使用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()]))
希望有人可以帮助我。
感谢与问候