luuyin / chase

Dynamic Sparsity Is Channel-Level Sparsity Learner [Neurips 2023]

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

How to test the flops, latency, and throughput of the pruned CIFAR-10/100 model?

dywu98 opened this issue · comments

Thanks for your amazing work! I am really interested in your work and trying to reproduce the results. Could you please provide an instruction on how to evaluate the flops, latency and throughput for CIFAR? I have run the command of pruning vgg19 on Cifar100 but found that there are only the final accuracy provided. How can I evaluate these metrics, especially the theoretical FLOPs and the GPU-supported FLOPs of the saved model? Is there any scripts for a quick evaluation?

Besides, how to reproduce the Chase+GraSP and Chase+SNIP method in the Table 3 in the paper? Which file do I have to run to reproduce them and how to assign the args?