Pruning with Uneven Block Dimensions

[arkadaw9]

Hi,

We just came across your work, it was amazing!

We have a question about the ResNet pruning that when we run the ChipNet code and get these results.

Test Acc: 0.7316
Total Trainable Params: 317198
Channels pruned: 80.00%
Parameters pruned: 81.63%

Test Acc: 0.7556
Total Trainable Params: 853033
Channels pruned: 60.00%
Parameters pruned: 50.59%

We saw that the output channels for the blocks are not the same, so how can we get a smaller model with it?
Did you do padding to make sure they are the same?