How to calculate GFlops correctly

Sense-X / UniFormer

[ICLR2022] official implementation of UniFormer

Apache License 2.0

817 stars 111 forks source link

Closed LEM0NTE closed 1 year ago

LEM0NTE commented 1 year ago

哈喽，再...再次感谢您的工作，但我遇到了一个新的问题。我注意到您提供的s8*8_K400模型的计算复杂度为70G 但是在我尝试训练模型时，misc.py: 188: 17.613435904 G 为什么这两个值不对应呢，log文件显示的这个17.6G代表的是什么计算复杂度呢

W4ngH4o commented 1 year ago

8x1x1: 17.6G 8x1x4: 70G(17.6x4)

Andy1621 commented 1 year ago

@W4ngH4o You are right. The correct FLOPs should be multiplied by #crop and #clip.