yolov5_prune_sfp issues

自己剪枝

1

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

suhanling

有几个问题想请教一下

2

作者你好，很感谢你开源代码，感谢，我有几个问题想请教一下： 1、您给出的模型是基于yolov5l训练的吧，可是为什么我剪枝后的模型要比我剪枝前的模型还要大呢？（剪枝前356.45M，剪枝后362.25M） 2、推理的时候可以用官方的detect.py来进行推理的吧

HandsLing

RuntimeError: Error(s) in loading state_dict for Small_Model: size mismatch for backbone_self.conv1.conv.weight: copying a param with shape torch.Size([104, 64, 3, 3]) from checkpoint, the shape in current model is torch.Size([89, 64,...

chuliuT

你好 File "D:\Anaconda3\envs\y55\lib\site-packages\torch\serialization.py", line 851, in _load result = unpickler.load() ModuleNotFoundError: No module named 'small_model_mod'

1

File "D:\Anaconda3\envs\y55\lib\site-packages\torch\serialization.py", line 851, in _load result = unpickler.load() ModuleNotFoundError: No module named 'small_model_mod'

ahhhxxx