Hassan686

> > 您的代码在train_prune_sfp中的m.init_mask(0.7, 3, 202, True)这一行设定了剪枝最深到202层，202之后的模型neck部分不进行剪枝，但是在get_small_model.py中small_model构建的时候neck却要剪枝：self.neck_self = neck(prune_rate)，这样模型维度不一致导致执行small_model.load_state_dict(small_state_dict)这一行时报错 > > 请问这是在运行哪个代码报的错？ 运行get_small_model.py时候会报错，ncek的维度不匹配

> > @xhwNobody > > 需要对代码进行修改，统一small_model网络结构和修剪后参数的维度 请问您能展开说说么，我也遇到了这样的问题