batchnorm-pruning

batchnorm-pruning copied to clipboard

thank for your codes. but I can not find any code that implements the process “absorb constant channels into next layer”, does it implement or not ? if done, can...

KarsonL

heerduo

TypeError: ResNet18Compressed() missing 1 required positional argument: 'channels'

2

hi, I run main.py, but an erro occured， what's wrong?thanks very much;

LanWong1

I found that the function "expand_model" may cause "out of memory" for a network with huge parameters.

JHsunshine

in paper, the ista algorithm described as rl+1 = max{(rl-u▽l),0}*sgn(rl-u▽l) , but in sgd.py, the ista algorithm described as: x = p.data.add(-group['lr'],d_p) x = torch.clamp((torch.abs(x) - ista), min=0.) p.data =...

talebolano

I replaced the resnet with the VGG and still had the same problem. In the sixth step, finetune's new model went wrong.The number of channels after pruning does not match...

luluvi

When running to the sixth step of finetune, there will be inconsistencies in the tensor dimension. How can I solve this problem? I look forward to your reply. RuntimeError: The...

luluvi

jack-willturner

jack-willturner