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