search

VITA-Group / Structure-LTH

[ICML 2022] "Coarsening the Granularity: Towards Structurally Sparse Lottery Tickets" by Tianlong Chen, Xuxi Chen, Xiaolong Ma, Yanzhi Wang, Zhangyang Wang.
MIT License
31 stars 6 forks source link

How can I recompile shmetis? #1

Open haodonga opened 1 year ago

haodonga commented 1 year ago

I see that the project provides shmetis for MACOS compilation, but my installation of metis in Linux does not contain shmetis, could the author please explain what shmetis is or give a way to recompile it.

xxchenxx commented 1 year ago

Hi, thanks for the interest in our project! The official release of hmetis contains shmetis which could be found in this link: http://glaros.dtc.umn.edu/gkhome/metis/hmetis/overview

haodonga commented 1 year ago

Thanks for the pointers, I have successfully run shmetis, what determines the number of hypergraphs, on the first epoch of Res18 training I have divided 40 hypergraphs, when will it end?

xxchenxx commented 1 year ago

Sorry for the late reply.

The number of hypergraphs is the same as the number used in [1]. We did not modify this part.

The process can be time-consuming, but it will end (less than 10 minutes).

Please let me know if you have any further question. Thanks!

[1] Accelerating Sparse CNN Inference on GPUs with Performance-Aware Weight Pruning