network-slimming

network-slimming copied to clipboard

如何进行迭代剪枝

10

文中所提到的迭代剪枝是将您提供的代码进行一次，然后将得到微调后的模型重复train with sparsity　，prune等过程吗？我实际去做的时候发现剪枝一次后的模型再重复剪枝的话减掉的通道和第一次剪枝一样，并没有剪枝掉更多的通道，这是什么原因呢？

liulonghoi

请问经稀疏训练剪枝后模型变小，但是我refine微调剪枝后的模型之后，模型又变大了，是什么原因导致？

ttttttt12

原论文中的memory是指

6

原论文中的表6中的memory指的是什么？我训练出的vgg16的权重文件只有118MB,可表6中的baseline的memory为697MB，这两个不是一个东西吗

PQ-girl

剪枝后保存的权重文件和newmodel加载不上

3

您好，我剪枝resnet164后保存为pruned.pth.tar和新的cfg，但是只用newmodel加载的时候无法加载pruned.pth.tar，报错为参数尺寸不匹配，想请教您这是因为channel_selection没有处理好吗?

lovegit2021

请问为何微调之后保存的模型大小变大了

4

您好，首先感谢您的工作 如题，我打印出来了剪枝之后的模型参数和微调之后的模型参数，发现微调之后模型的channel又恢复到了原模型每个卷积层的channel，所以微调之后保存的模型就变大了，不知道是我的操作问题还是代码就是这个样子的呢

wayne2tech

mnn加载剪枝模型错误

1

剪枝后进行mnn转换，报错 Error for compute convolution shape, need channel = 62, input channel = 64 Error for compute convolution shape, need channel = 62, input channel = 64 Error for compute...

DLTensor

TypeError: item() takes no arguments (1 given)

1

运行main.py时报错 100. * batch_idx / len(train_loader), loss.item(0))) TypeError: item() takes no arguments (1 given)

Rhett-Lin

预测单张照片时，准确率只有0.0几

2

你好，我想问一下我将剪枝后的模型进行微调，微调后准确率恢复到90几，后加载微调后的模型进行预测单张图片，结果都是错误的，请问是什么原因呢

PQ-girl

mask-impl fine-tune

2

你好，mask版本的prune_mask并没有真正对权重进行剪枝，在finetune时，虽然用的是mask后的权重，但是对所有parameters进行更新，这样bn那些为0的权重不是也被训练了吗？

colorjam

If the remaining channel for a layer is zero, it reports zero division error

2

I trained vgg with cifar100 dataset and there are a few layers with zero remaining channels. As a result I got a zerodivisionerror in vgg.py part when initializing weights. Can...

Neil-untitled