STEP3:通道剪枝策略一模型加载不了

ZJU-lishuang / yolov5_prune

yolov5 prune，Support V2, V3, V4 and V6 versions of yolov5

Apache License 2.0

556 stars 136 forks source link

Closed ztrong-forever closed 2 years ago

ztrong-forever commented 2 years ago

您好大佬！我将剪枝后的模型保存后，想在val.py中加载模型测试效果，可是一直加载不了

ZJU-lishuang commented 2 years ago

参考该做法剪枝模型保存的不是yolov5格式的模型，需要转一下

ztrong-forever commented 2 years ago

好的，感谢答复！！