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.)
