我正在尝试对自定义模型预测例程进行一些图像批量预测。我已按照本教程创建例程包装器。我已经在谷歌云AI平台上将我的模型部署为tar.gz创建了一个版本等等。现在,当我尝试使用此命令从命令行启动批量预测时:
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,它会直接显示它包含的图像。很酷! 但我不明白这个错误。
批量预测目前不支持自定义预测。它仅支持在线预测。错误消息应该只是显示:"批量预测仅支持Tensorflow。