Markkic
Results
3
comments of
Markkic
I got it. just change the image size from 160 to 640(or other size)
> > 您的代码在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)这一行时报错 > > 请问这是在运行哪个代码报的错? small_model_mod.py的29行:self.neck_self = neck(self.prune_rate) 请问各位有解决这个问题的办法吗?
是啊,过期了