No handlers found "prim::pythonop" and "prim::tupleunpack"

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

Closed Cosijopiii closed 3 years ago

Cosijopiii commented 3 years ago

hope to support "prim::pythonop" and "prim::tupleunpack" operations

Thanks and great tool!

zhijian-liu commented 3 years ago

Thanks! Could you please provide a sample code to reproduce this?