GitHub repository: HaoKang-Timmy/pytorch-OpCounter
Primary language: Python
Description: Count the MACs / FLOPs of your PyTorch model.