search

zhijian-liu / torchprofile

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

No handlers found: "aten::zeros" and "aten::lstm" Skipped. #4

Open un-knight opened 4 years ago

un-knight commented 4 years ago

hope to support "aten::zeros" and "aten::lstm" ops, thanks!

zhijian-liu commented 4 years ago

"aten::zeros" does not require any mult-adds and therefore will not affect the final result. I'm currently working on supporting "aten::rnn" and "aten::lstm" and will let you know once they are supported.

un-knight commented 4 years ago

Thanks! that would be really cool.

zwlanpishu commented 2 years ago

"aten::zeros" does not require any mult-adds and therefore will not affect the final result. I'm currently working on supporting "aten::rnn" and "aten::lstm" and will let you know once they are supported.

Does this tool support pytorch models like RNNs and some customized modules? If a model is not supported, will the tool output an error message or an inaccurate result. Thank you very much for any guidance.

zhijian-liu commented 2 years ago

It currently does not support RNNs. You will see a warning if the module is not supported.