search

midasklr / yolov5prune

548 stars 113 forks source link

使用train_sparsity.py出现的错误- #131

Open liujialong2019 opened 1 year ago

liujialong2019 commented 1 year ago

AttributeError: Can't get attribute 'DetectionModel' on <model.yolo' from 'path/model/yolo.py'>

laoyu-jidianle commented 1 year ago

你用了新版本的模型，可以在Yolov5的git上面，下v6.0版本的模型文件