pytorch-pruning
pytorch-pruning copied to clipboard
jacobgil
PyTorch Implementation of [1611.06440] Pruning Convolutional Neural Networks for Resource Efficient Inference
hi jacobgil It is a very nice work, Just wondering if you have any ideas or suggestions for Batch normalization which contains the info from upstream conv layer ? We...
Hi, first of all thanks for the awesome work. i'm trying ti utilize your work to a more complicated model (xception). in this model there are conv layers with groups...
The given example of VGG16 works well for me. I want to try out this method on a MobiletNetV2 model. VGG16 has a pretty simple structure: a sequence of convs...
Hi, How long should I expect one pruning iteration to take? I'm talking about one iteration in this loop: for layer_index, filter_index in prune_targets: model = prune_vgg16_conv_layer(model, layer_index, filter_index) It...
Hi I have encountered this error while running the code. After getting the information about the filters which are to be pruned, while pruning the filters this issue occured. ```...
What could be the cause and fix of this CUDNN_STATUS_BAD_PARAM error? I cannot get this to extract the features. Pytorch otherwise runs fine on my Ubuntu + Anaconda machine. >my@my:~/Dropbox/x/CV/pytorch-pruning$...
My running environment is: - Ubuntu 16.04LTS - pytorch 0.4.1 - python 3.6 I got a problem is: `/home/wangshuo/anaconda3/envs/pytorch/lib/python3.6/site-packages/torchvision-0.2.1-py3.6.egg/torchvision/transforms/transforms.py:188: UserWarning: The use of the transforms.Scale transform is deprecated, please use...
error dimension out of range values = \ torch.sum((activation * grad), dim = 0). \ sum(dim=2).sum(dim=3)[0, :, 0, 0].data
When I use [my own modified pytorch-pruning for python 3](https://github.com/promach/pytorch-pruning/) with [this fruits dataset](https://github.com/Horea94/Fruit-Images-Dataset) , I have quite low accuracy value. May I know why ? Besides, how do we...
