pytorch-utils
Flops counter for neural networks in pytorch framework
🤓🔥 Collection of PyTorch additions, extensions, utilities, uses and abuses
Seamless analysis of your PyTorch models (RAM usage, FLOPs, MACs, receptive field, etc.)
