microsoft / Cream

This is a collection of our NAS and Vision Transformer work.
MIT License
1.61k stars 220 forks source link

EfficientVIT'pie charts #228

Open YEKAI-2022 opened 3 months ago

YEKAI-2022 commented 3 months ago

Hello, I would like to know how the SwinT and DeiT pie charts that limit memory operations in the EfficientVit paper are analyzed. Can you provide the source code?

xinyuliu-jeffrey commented 3 months ago

Hi @YEKAI-2022 , thanks for your interest!

We use torch-profile link and nsight-compute link for runtime profiling. Please refer to the links for more details. Thanks.