为了方便实现，这个repo里的剪枝是通过给卷积层输出乘以一个mask实现的等效剪枝，没有真正把卷积核去掉，所以内存没有变化
Opened this issue · 0 comments
xu00123 commented
为了方便实现，这个repo里的剪枝是通过给卷积层输出乘以一个mask实现的等效剪枝，没有真正把卷积核去掉，所以内存没有变化
如果需要减小内存只要把mask为0对应的卷积核去掉就可以，你可以自己实现一下
Originally posted by @zzzxxxttt in #2 (comment)