A tool to count the FLOPs of PyTorch model.
[](/packages/thop)
<a href="/packages/thop"><img src="/api/badges/thop?period=month" alt="PyPI Stats"></a>