1 Resposta
- Mais recentes
- Mais votos
- Mais comentários
0
【以下的回答经过翻译处理】 是的,从这个github的issue中可以看到:https://github.com/aws/sagemaker-pytorch-inference-toolkit/issues/86
在这个issue中指出,新的Pytorch 1.6镜像要求模型文件名为“model.pth”,并链接到相关代码,其中设置了此默认值:https://github.com/aws/sagemaker-pytorch-inference-toolkit/blob/9a6869e/src/sagemaker_pytorch_serving_container/torchserve.py#L121
该线程中还指出,用户已经成功地适应了他们的代码来使用Pytorch 1.6中的torchserve,通过将其更改为在名为“model.pth”的文件中保存他们的模型。一旦重命名,他们仍然可以使用自定义推断脚本通过定义自定义“model_fn”来加载其模型:https://github.com/data-science-on-aws/workshop/blob/374329adf15bf1810bfc4a9e73501ee5d3b4e0f5/09_deploy/wip/pytorch/code/inference.py
Conteúdo relevante
- AWS OFICIALAtualizada há 2 anos
- AWS OFICIALAtualizada há 2 anos
- AWS OFICIALAtualizada há 2 anos