Fix ProtoPNet pruning
ProtoPNet pruning should be followed by a fine-tuning step on the last (linear) layer. In the current code version, the drop in accuracy after pruning is almost 8% on CUB200.
ProtoPNet pruning should be followed by a fine-tuning step on the last (linear) layer. In the current code version, the drop in accuracy after pruning is almost 8% on CUB200.