【发布时间】: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
?