GitHub repository: zhijian-liu/torchprofile
Primary language: Python
Description: A general and accurate MACs / FLOPs profiler for PyTorch models