Pruning time consumption

[Li-Whasaka]

Step 2: Prune the models (SECOND as example, on 8 GPUs)： How long does it take for you to fall on the dataset once during the pruning iteration process?

[zhnxjtu]

Hi,

Calculating the pruning measure will take some time (not very short), as this step requires iterating over all samples in the dataset. Here, I provide a couple of possible ways to reduce the time consumption:

• Increase the batch size as much as possible;
• Randomly select a subset of the dataset for calculating the measure (similar to HRank [1]).

Additionally, the measure result for SECOND is available at "tools/measure_doc/SECONDNet_kitti.txt".

[1] Lin M, Ji R, Wang Y, et al. Hrank: Filter pruning using high-rank feature map[C], CVPR 2020.

Best regards.