在 Sagemaker Inference 中找不到模型路径

Question

我正在尝试执行与姿势相关的活动作为推理的一部分。我正在使用 Fastapi 在 Sagemaker 上执行推理，方法如下 https://sii.ua/blog/en/deploying-custom-models-on-aws-sagemaker-using-fastapi/

我这里有一堆预模型，我将它们作为 tar 文件存储在 S3 上。 Cmd 用于 tar 文件：

cd opt/ml/
tar -czvf pre-models.tar.gz pre-models

在我的代码中，我将模型路径设置为“ml/pre_models/（子目录根据要求而变化）” 但是，Sagemaker 找不到模型路径。我在 cloudwatch 中看到这一行：

“修复模型/ ml/pre_models/models 的路径”

我尝试了 ls，但看不到其中的预模型文件。

Answer 1

为了降落在这里的未来的自己...

我的代码位于 /opt/ml/ 中，模型位于 pre_models/

中

Sagemaker Inference 创建目录 model/ 来存储 S3 模型。所以最终的路径显示为：

/opt/ml/model/pre_models

要跟踪此路径，只需继续从代码中打印目录结构（在我的例子中为“调用”API 方法

print(os.system("ls"))
print(os.system("ls /opt/ml"))