how Pruning the last conv layer affects the first linear layer of the classifier
Saharkakavand opened this issue · 0 comments
Saharkakavand commented
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.