как
аюсь развернуть переобученную версию начальной модели на google cloud ml-engine. Сбор информации отДокументация SavedModel, этотссылка, и этосообщение из rhaertel80 я успешно экспортировал свою переподготовленную модель в SavedModel, загрузил ее в корзину и попытался развернуть в версии ml-engine.
Эта последняя задача на самом деле создает версию, но выводит эту ошибку:
Create Version failed. Bad model detected with error: "Error loading the model: Unexpected error when loading the model"
И когда я пытаюсь получить прогнозы от модели через командную строку, я получаю это сообщение об ошибке:"message": "Field: name Error: Online prediction is unavailable for this version. Please verify that CreateVersion has completed successfully."
Я сделал несколько попыток, пытаясь разныеmethod_name
а такжеtag
варианты но ни один не работал.
Код, добавленный к исходному начальному коду:
### DEFINE SAVED MODEL SIGNATURE
in_image = graph.get_tensor_by_name('DecodeJpeg/contents:0')
inputs = {'image_bytes': tf.saved_model.utils.build_tensor_info(in_image)}
out_classes = graph.get_tensor_by_name('final_result:0')
outputs = {'prediction': tf.saved_model.utils.build_tensor_info(out_classes)}
signature = tf.saved_model.signature_def_utils.build_signature_def(
inputs=inputs,
outputs=outputs,
method_name='tensorflow/serving/predict'
)
### SAVE OUT THE MODEL
b = saved_model_builder.SavedModelBuilder('new_export_dir')
b.add_meta_graph_and_variables(sess,
[tf.saved_model.tag_constants.SERVING],
signature_def_map={'predict_images': signature})
b.save()
Еще одно соображение, которое может помочь: я использовал экспортированныйtrained_graph.pb
сgraph_def.SerializeToString()
чтобы получить прогнозы локально, и он работает нормально, но когда я заменяю егоsaved_model.pb
это терпит неудачу.
Любые предложения о том, что проблема может быть?