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

Lyken17 / pytorch-OpCounter

Count the MACs / FLOPs of your PyTorch model.

MIT License

4.85k stars 528 forks source link

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

Open cmc370 opened 11 months ago

cmc370 commented 11 months ago

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.