我一直在尝试使用YoloV5 Neo编译我的自定义培训的SageMaker模型。编译给出一个错误:
ClientError: InputConfiguration: No pth file found for PyTorch model.
Please make sure the framework you select is correct.权重是一个.pt文件。有办法将.pt文件转换为.pth吗?
发布于 2021-12-29 10:37:33
这两个文件扩展名之间没有区别。如果SageMaker只需要一个扩展名为.pth的文件,那么您可以简单地将文件从filename.pt重命名为filename.pth。
https://stackoverflow.com/questions/70508395
复制相似问题