Модель нагрузки Tensorflow в 1.11 - PullRequest
Новая
       26

Модель нагрузки Tensorflow в 1.11

0 голосов
/ 20 июня 2020

Я хочу использовать 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 мне нужно было только указать путь, и все было хорошо. Теперь мне нужен сеанс и тег, но я понятия не имею, что я должен указать в качестве тега, я никогда его не устанавливал. 

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()]))

Надеюсь, кто-нибудь может мне помочь.

Спасибо и С уважением

...