Google Cloud ML Engine: проблема с локальным прогнозом, учитывая сохраненный_модель.pb - PullRequest
Новая
       25

Google Cloud ML Engine: проблема с локальным прогнозом, учитывая сохраненный_модель.pb

0 голосов
/ 23 октября 2018

Я обучил модель Keras с использованием tf.data.Dataset API и пытаюсь проверить, правильно ли я сохранил ее (как saved_model.pb), чтобы я мог использовать ее на ML Engine.Вот что я сделал: 

estimator = tf.keras.estimator.model_to_estimator(my_model)
# create serving function...
estimator.export_savedmodel('./export', serving_fn)

Так что теперь я пытаюсь использовать gcloud ml-engine local predict, чтобы посмотреть, смогу ли я получить прогноз обратно.Я делаю: 

gcloud ml-engine local predict --model-dir=~/path/to/folder --json-instances=instances.json

К сожалению, я получаю: 

cloud.ml.prediction.prediction_utils.PredictionError: Failed to load model: Cloud ML only supports TF 1.0 or above and models saved in SavedModel format. (Error code: 0)

Итак, я пытаюсь добавить --runtime-version=1.2 к моей команде так: 

gcloud ml-engine local predict --model-dir=~/path/to/folder --json-instances=instances.json --runtime-version=1.2

и я возвращаюсь: 

ERROR: (gcloud.ml-engine.local.predict) unrecognized arguments: --runtime-version=1.2

Есть идеи, что я делаю неправильно / как исправить?

Спасибо!

1 Ответ

0 голосов
/ 01 ноября 2018

Для потомков: проблема оказалась в неверном пути.Если кто-то еще сталкивается с этой проблемой, попробуйте использовать полный абсолютный путь и убедитесь, что вы указываете на каталог, содержащий файл saved_model.pb.

...