SageMaker PyTorch Hosting 1.6 只能使用名为model.pth的模型文件吗？

Question

【以下的问题经过翻译处理】 嗨，SageMaker PyTorch Hosting 1.6 只与名为 model.pth 的艺术品一起工作吗？
我正在尝试使用 1.6 的这个示例（https://github.com/aws-samples/amazon-sagemaker-bert-classify-pytorch），但部署失败并显示错误消息：

```
FileNotFoundError: [Errno 2] No such file or directory: '/opt/ml/model/model.pth'

```

文档中没有提到这样的限制。

Answer

【以下的回答经过翻译处理】 是的，从这个github的issue中可以看到：

在这个issue中指出，新的Pytorch 1.6镜像要求模型文件名为“model.pth”，并链接到相关代码，其中设置了此默认值：

该线程中还指出，用户已经成功地适应了他们的代码来使用Pytorch 1.6中的torchserve，通过将其更改为在名为“model.pth”的文件中保存他们的模型。一旦重命名，他们仍然可以使用自定义推断脚本通过定义自定义“model_fn”来加载其模型：

SageMaker PyTorch Hosting 1.6 只能使用名为model.pth的模型文件吗？

相关内容