Count the MACs / FLOPs of your PyTorch model. - View it on GitHub