将 .pt 转换为 .pth？

Question

我一直在尝试使用 SageMaker Neo 编译我经过自定义训练的 YoloV5 模型。 编译报错：

ClientError: InputConfiguration: No pth file found for PyTorch model. 
Please make sure the framework you select is correct.

权重是.pt 文件。 有没有办法将.pt 文件转换为.pth？

Answer 1

两个文件扩展名之间没有区别。如果 SageMaker 只查找扩展名为 .pth 的文件，那么您只需将文件从 filename.pt 重命名为 filename.pth。