将 .pt 转换为 .pth?

分享于2022年07月17日 amazon-sagemaker python pytorch yolov5 问答
【问题标题】:将 .pt 转换为 .pth?(Convert .pt to .pth?)
【发布时间】:2022-01-27 04:20:22
【问题描述】:

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

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

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


【解决方案1】:

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

  • 这似乎不起作用,我收到错误消息 ClientError: InputConfiguration: Framework cannot load PyTorch model. No module named 'models'
  • 我认为这是一个不相关的错误