pytorch-pruning
pytorch-pruning copied to clipboard
jacobgil
PyTorch Implementation of [1611.06440] Pruning Convolutional Neural Networks for Resource Efficient Inference
Hi, Jacob, Seems the python script's indentation mixed tabs and spaces.
I use Tesla k80 -12G *4,When I prunning the training test everything was normal, but after the pruned test memory overflow. THCudaCheck FAIL file=/pytorch/aten/src/THC/generic/THCStorage.cu line=58 error=2 : out of memory...
the speed of model pruned did not improve, how to your work?
I run python finetune.py --train the output accuracy is always 0 I use the dataset Dogs vs. Cats
In finetune.py, why do you have two different classes : FilterPrunner and PrunningFineTuner_VGG16 ? Isn't it possible to merge those two classes to have only one class ? Thank you
Hi there, just wanted to say thank you for the blog post and the code example. I noticed that the function `compute_rank` in finetune.py is mutating a global state, namely...
how to obtain VGG16: flops = [3.1, 57.8,14.1, 28.9, 7:0, 14.5,14.5,3.5, 7.2,7.2, 1.8,1.8, 1.8, 1.8] according to FLOPs = 2HW(CinK2 + 1)Cout? for example, no.1 conv FLOPs(0) = 3.1, while...
Could it was pruned to ResNet model? If ResNet model loaded, it occured error to FilterPrunner.
Excuse me, about gradient I have some not understand, why the gradient shapes same as activation output shape. And the gradient is not weight gradient? it shape is [i ,...
Hi Jacob! Thanks for your impressive work. I ran your code pruning VGG16 by removing 60% filters. However, the runtime for testing only speeds up from 0.64sec/image to 0.39sec/image, which...
