Torch-Pruning icon indicating copy to clipboard operation
Torch-Pruning copied to clipboard

关于yolov8的稀疏训练

Open TnoobT opened this issue 9 months ago • 6 comments

我的剪枝流程是 1. 先提前测试验证集精度。 2. 稀疏训练10个epoch。 (3.剪枝。 4. 验证剪枝后的精度。 5. finetune并验证精度。重复这三步) 我有以下几个问题:

  1. 我这剪枝流程是否正确
  2. 在finetune的时候,是否要加入pruner.regularize(self.model)
  3. pruner中的参数如iterative_steps和ch_sparsity这些会不会影响稀疏训练,因为这里的iterative_steps=1,而我总的iterative_steps=16。这些参数会对稀疏训练有影响嘛。 期待大佬回答 :> image

TnoobT avatar Oct 12 '23 05:10 TnoobT

@VainF 大佬求回复

TnoobT avatar Oct 16 '23 02:10 TnoobT

您好,

  1. 没问题
  2. 不用加入pruner.regularize(self.model)
  3. 会有影响,需要自己调调参

VainF avatar Oct 16 '23 05:10 VainF

@VainF 大佬,这是我测试剪枝的结果。https://blog.csdn.net/qq_33596242/article/details/133774348 ,您能帮我看看正常不。正常的话,我这流程也可以给其他人做个参考

TnoobT avatar Oct 24 '23 01:10 TnoobT

@VainF 大佬,这是我测试剪枝的结果。https://blog.csdn.net/qq_33596242/article/details/133774348 ,您能帮我看看正常不。正常的话,我这流程也可以给其他人做个参考

结果是正常的,我是他学弟,他自己没完整测过yolo的剪枝

HollyLee2000 avatar Dec 14 '23 09:12 HollyLee2000

@Reaidu 用我博客里的版本

TnoobT avatar Dec 26 '23 08:12 TnoobT

结果是正常的,我是他学弟,他自己没完整测过yolo的剪枝

师兄你好,请问组剪枝的原理是什么,原论文没有太理解,呜呜呜 1705381890161 感谢感谢!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!

Reaidu avatar Jan 16 '24 05:01 Reaidu