Closed tangchen2 closed 3 years ago
prune_utils.py中的prune_model_keep_size,pruned_model似乎只是将原来记录的pruned param给赋值过去了,仍然是256,512这些维度,mask=0的通道在模型中仍然存在,并没有真正的剪掉?那么模型的尺寸就不会缩小了吧?
代码中是通过修改cfg的方式重新生成剪枝后的model的,pruned model只是将权重置零,compact model才是真正剪枝的model close issue
prune_utils.py中的prune_model_keep_size,pruned_model似乎只是将原来记录的pruned param给赋值过去了,仍然是256,512这些维度,mask=0的通道在模型中仍然存在,并没有真正的剪掉?那么模型的尺寸就不会缩小了吧?