首页
学习
活动
专区
圈层
工具
发布
社区首页 >问答首页 >将.pt转换为.pth?

将.pt转换为.pth?
EN

Stack Overflow用户
提问于 2021-12-28 14:25:11
回答 1查看 753关注 0票数 0

我一直在尝试使用YoloV5 Neo编译我的自定义培训的SageMaker模型。编译给出一个错误:

代码语言:javascript
复制
ClientError: InputConfiguration: No pth file found for PyTorch model. 
Please make sure the framework you select is correct.

权重是一个.pt文件。有办法将.pt文件转换为.pth吗?

EN

回答 1

Stack Overflow用户

发布于 2021-12-29 10:37:33

这两个文件扩展名之间没有区别。如果SageMaker只需要一个扩展名为.pth的文件,那么您可以简单地将文件从filename.pt重命名为filename.pth

票数 -1
EN
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/70508395

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档