search

lmbxmu / HRankPlus

An extension version of our paper accepted by CVPR 2020, Oral -- HRank: Filter Pruning using High-Rank Feature Map
https://128.84.21.199/abs/2002.10179
143 stars 21 forks source link

How the selected_index is used in the load_densenet_model function? #11

Closed pvti closed 1 year ago

pvti commented 1 year ago

Hi, Thanks for the repository. From my understanding, in this function: https://github.com/lmbxmu/HRankPlus/blob/bcdb51dd7f039d8fa295eeafa7bc78bc7dab3d75/evaluate_cifar.py#L511 The selected_index isn't used to update to input channels of the next layer. Can you confirm this?

pvti commented 1 year ago

I've figured out: https://github.com/lmbxmu/HRankPlus/blob/bcdb51dd7f039d8fa295eeafa7bc78bc7dab3d75/evaluate_cifar.py#L561