Skip to content

Files

Latest commit

1e8bb8b · May 29, 2021

History

History
7 lines (6 loc) · 279 Bytes

README.md

File metadata and controls

7 lines (6 loc) · 279 Bytes

InteligenceBasePruner

a nn model pruner base on PSO 基于粒子群算法的神经网络结构化剪枝

  • 粒子群算法:搜索网络结构
  • 快速模型评估:只训练BN层
  • 压缩搜索空间:可以设置剪枝因子g,搜索空间是这个剪枝因子g整数倍