Я пытаюсь сделать некоторые предсказания пакета изображений для пользовательской процедуры прогнозирования модели.Я следовал этому руководству по созданию обычной обертки.Я развернул свою модель как tar.gz на облачной платформе Google, создал версию и так далее.Теперь, когда я пытаюсь запустить прогнозирование партии из командной строки с помощью этой команды:
gcloud ai-platform jobs submit prediction $JOB_NAME --model $MODEL_NAME \
--input-paths $INPUT_PATHS \
--output-path $OUTPUT_PATH \
--region $REGION \
--data-format $DATA_FORMAT
я получаю следующую ошибку
ERROR: (gcloud.ai-platform.jobs.submit.prediction) INVALID_ARGUMENT: Field: prediction_input.framework Error: Batch prediction only supports Tensorflow.FRAMEWORK_CUSTOM_CLASS
- '@type': type.googleapis.com/google.rpc.BadRequest
fieldViolations:
- description: Batch prediction only supports Tensorflow.FRAMEWORK_CUSTOM_CLASS
field: prediction_input.framework
Ввод в формате JSON, и я думаю, что этоправильно структурирован, потому что в ведре, на консоли, под типом написано {"instances": [{"image_bytes": {"b64": "image/jpeg"}}]}
, и если я нажму на JSON, он покажет прямое изображение, которое в нем содержится.Довольно круто!Но я не понимаю ошибку.