Pytorch Utils Python Packages

ptflops

Flops counter for neural networks in pytorch framework

ebtorch

🤓🔥 Collection of PyTorch additions, extensions, utilities, uses and abuses

torchscan

Seamless analysis of your PyTorch models (RAM usage, FLOPs, MACs, receptive field, etc.)