I am trying to use prune.py in python 3.7.1 and pytorch 1.0, but I have the following error:
[phung@archlinux SqueezeNet-Pruning]$ python finetune.py --prune
/usr/lib/python3.7/site-packages/torchvision/transforms/transforms.py:187: UserWarning: The use of the transforms.Scale transform is deprecated, please use transforms.Resize instead.
warnings.warn("The use of the transforms.Scale transform is deprecated, " +
/usr/lib/python3.7/site-packages/torchvision/transforms/transforms.py:562: UserWarning: The use of the transforms.RandomSizedCrop transform is deprecated, please use transforms.RandomResizedCrop instead.
warnings.warn("The use of the transforms.RandomSizedCrop transform is deprecated, " +
Accuracy : 1.0
Number of prunning iterations to reduce 67% filters 13
Ranking filters…
Layers that will be prunned {3: 2, 6: 9, 12: 77, 9: 10, 10: 8, 11: 14, 7: 7, 0: 1}
Prunning filters…
Traceback (most recent call last):
File “finetune.py”, line 306, in
fine_tuner.prune()
File “finetune.py”, line 242, in prune
model = prune_squeezenet_conv_layer(model, layer_index, filter_index)
File “/home/phung/Documents/Grive/Personal/Coursera/Machine_Learning/pruning/Pruning-CNN/SqueezeNet-Pruning/prune.py”, line 146, in prune_squeezenet_conv_layer
torch.nn.Conv2d(in_channels = old_conv_layer.in_channels-1,
AttributeError: ‘NoneType’ object has no attribute ‘in_channels’
[phung@archlinux SqueezeNet-Pruning]$