search

zhijian-liu / torchprofile

A general and accurate MACs / FLOPs profiler for PyTorch models
https://pypi.org/project/torchprofile/
MIT License
571 stars 39 forks source link

Handle `aten::hardtanh` #14

Closed abcdabcd987 closed 3 years ago

abcdabcd987 commented 3 years ago

BTW, this is a very handy library. Huge thanks!