Auto_pruning
Auto_pruning copied to clipboard
auto Iterative pruning based on caffe
将原来的Convolution改成CConvolution,再加上convolution_param后,用预训练模型训练,就报了这个错误,是我哪里没做对吗,求助
有下面几个问题: 1)compress_conv_layer.cu中Forward_gpu中关于weightMaskCPU不应该重复全部设置1 2)weightMaskCPU的数据中不是0,就是1,但是在训练过程中,数据不断降低,这里数据不应该应用梯度下降
请问后续要怎么才能把模型变小呢?
help
Hi, Thank you very much for your open source code, how can the pruning model become smaller?Can you provide a small implementation example? thanks for your help.