自己剪枝

[suhanling]

博主 是不是按照你知乎上写的，将model.py small_model_mod.py中改网络参数结构改成自己所想剪枝的网络，然后配置好train.py，然后data/xxx.yaml 配置好，就可以了。getsmall.py是获取剪枝后的模型还是 train训练完之后保存的best.py就是剪枝后的模型呢。

[xhwNobody]

博主 是不是按照你知乎上写的，将model.py small_model_mod.py中改网络参数结构改成自己所想剪枝的网络，然后配置好train.py，然后data/xxx.yaml 配置好，就可以了。getsmall.py是获取剪枝后的模型还是 train训练完之后保存的best.py就是剪枝后的模型呢。

不是哈，这里已经将yolov5官网的xxx.yaml配置文件构建模型的方式换了，getsmall.py是将模型权重进行去0修剪