GitHub repository: saar-eliad/pytorch-OpCounter
Description: Count the MACs / FLOPs of your PyTorch model.