我正在嘗試從 Postman 和 AWS CLI 調用我的 SageMaker 模型終端節點。端點的狀態是“服務中”,但每當我嘗試調用它時,它都會給我一個錯誤。當我嘗試在 SageMaker notebook 中使用預測函數并為其提供一個 numpy 數組(例如np.array([1,2,3,4]))時,它成功地為我提供了一個輸出。我不確定我做錯了什么。$ aws2 sagemaker-runtime invoke-endpoint \$ --endpoint-name=pytorch-model \$ --body=1,2 \$ --content-type=text/csv \$ --cli-binary-format=raw-in-base64-out \$ output.jsonAn error occurred (ModelError) when calling the InvokeEndpoint operation: Received server error (500) from model with message "tensors used as indices must be long, byte or bool tensorsTraceback (most recent call last): File "/opt/conda/lib/python3.6/site-packages/sagemaker_inference/transformer.py", line 125, in transform result = self._transform_fn(self._model, input_data, content_type, accept) File "/opt/conda/lib/python3.6/site-packages/sagemaker_inference/transformer.py", line 215, in _default_transform_fn prediction = self._predict_fn(data, model) File "/opt/ml/model/code/pytorch-model-reco.py", line 268, in predict_fn return torch.argsort(- final_matrix[input_data, :], dim = 1)IndexError: tensors used as indices must be long, byte or bool tensors
為 PyTorch 模型調用 SageMaker 端點
慕的地8271018
2023-04-25 17:31:10