coldlarry / YOLOv3-complete-pruning

提供对YOLOv3及Tiny的多种剪枝版本,以适应不同的需求。
805 stars 214 forks source link

必须要进行基础训练才能进行稀疏训练吗? #83

Open luhang-HPU opened 4 years ago

luhang-HPU commented 4 years ago

如题。第一次基础训练yolo.weights生成了last.pt,之后进行稀疏训练的话,我看命令行还是--weights weights/yolov3.weights,是不是可以直接进行稀疏训练,不必先基础训练生成last.pt?

ZhihangTang commented 4 years ago

这个看你的任务吧,如果你的任务用原本的yolov3.weights效果就已经很好了,可以直接稀疏训练。如果需要进行迁移学习,那则要先基础训练。

nanhui69 commented 2 years ago

这个看你的任务吧,如果你的任务用原本的yolov3.weights效果就已经很好了,可以直接稀疏训练。如果需要进行迁移学习,那则要先基础训练。

“ 如果需要进行迁移学习,那则要先基础训练 “意思是 在这一步的稀疏训练 ,加载的模型是基础训练生成好的 last.pt么?