How to calculate FLOPs(G) and Param(M) in the code?

Vegetebird / StridedTransformer-Pose3D

[TMM 2022] Exploiting Temporal Contexts with Strided Transformer for 3D Human Pose Estimation

MIT License

335 stars 37 forks source link

How to calculate FLOPs(G) and Param(M) in the code? #39

Closed zhang-yuxian closed 7 months ago

zhang-yuxian commented 10 months ago

I used the ptflops library to calculate FLOPs and Param, but the FLOPs values differ significantly from the paper. So how do I calculate the FLOPs(G) and Param(M) in the code?

Vegetebird commented 7 months ago

We calculate the FLOPs manually. You could consider using thop package for calculation.