pytorch-pruning icon indicating copy to clipboard operation
pytorch-pruning copied to clipboard

how Pruning the last conv layer affects the first linear layer of the classifier

Open Saharkakavand opened this issue 5 years ago • 0 comments

I trained the vgg and saved the model as pth file. then I load it for pruning some filters of it. the last conv after pruning is not 512 anymore, some filters are gone. how Pruning the last conv layer affects the first linear layer of the classifier which is (512 7 7, 4096). how can I prune the input weights of classifier according to the last conv layer.

Saharkakavand avatar Jan 09 '20 11:01 Saharkakavand