sdc17 / UPop

[ICML 2023] UPop: Unified and Progressive Pruning for Compressing Vision-Language Transformers.
https://dachuanshi.com/UPop-Project/
BSD 3-Clause "New" or "Revised" License
96 stars 5 forks source link

您好,想问一下FLOPs指标的问题 #33

Closed zushiye1 closed 5 months ago

zushiye1 commented 5 months ago

请问,这些模型中的FLOPs指标是如何得到的呢

sdc17 commented 5 months ago

By fvcore package. See here https://github.com/sdc17/UPop/blob/6aae798a9a576cf001ab1ca27b5afc15cbeeda46/deit/main.py#L343-L350