pthflops (pypi): Estimate FLOPs of neural networks
Keywords: convolutional-neural-networks|deep-learning|flops|pytorch|pytorch-estimate-flops