Flops counter for neural networks in pytorch framework - View it on GitHub