Count the MACs / FLOPs of your PyTorch model.
Geek Repo:Geek Repo
Github PK Tool:Github PK Tool
cmc370 opened this issue 9 months ago · 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.