vegetable-chao / torchprofile Goto Github PK
View Code? Open in Web Editor NEWThis project forked from zhijian-liu/torchprofile
A general and accurate MACs / FLOPs profiler for PyTorch models
Home Page: https://pypi.org/project/torchprofile/
License: MIT License