Lyken17 / pytorch-OpCounter

Count the MACs / FLOPs of your PyTorch model.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Is thop also effective for calculating Flops for spiking neural networks?

cmc370 opened this issue · comments

I'm currently conducting research on spiking neural networks and would like to measure their energy consumption on GPUs. However, I'm not sure whether thop supports this.