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