在YOLOv5训练自己的数据集模型时删除预训练权重后发现报错

在YOLOv5训练自己的数据集模型时删除预训练权重后发现报错_第1张图片

 上图是报错内容,

找到general.py 出错的位置,应该是YOLOv5版本的问题,就用一个可以正常空权重跑通的文件,将general.py相应部分给复制下来,粘贴过去。

报错内容是:.acceptable suffix is ['.pt']

你可能感兴趣的:(深度学习,人工智能,神经网络)