question about flops - Githubissues

EasonXiao-888 / GrootVL

The official implementation of GrootVL: Tree Topology is All You Need in State Space Model

58 stars 2 forks source link

question about flops #7

Open zhouyuan888888 opened 2 months ago

zhouyuan888888 commented 2 months ago

Hi, thank you for your great work and it is very interesting! But I have a question, how can I calculate the FLOPs of the whole model? Can you provide a script about calculating flops?