-
Notifications
You must be signed in to change notification settings - Fork 2.8k
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
(feature) Ops estimation per node/per net. #107
Comments
What do you mean by ops? Can you provide some examples? This would require #71 to be addressed first. The tool doesn't have a dependency on the Python runtime and frameworks. |
I was thinking of something along the lines of https://arxiv.org/abs/1605.07678, where the paper ranks neural nets based on the Accuracy (vertical axis, bubble chart) and G-Operations. The GOPs number indicates how many operations (usually multiplications+additions) are spent on a particular layer. This tool provides a reference calculation of OPs for different layers: https://github.com/apaszke/torch-opCounter/tree/master/src (see profiler.lua). Thanks for considering these feature requests 👍 |
Would be great to understand how many ops are required per each node and overall in the network (maybe under the assumption of an acyclical graph).
This can actually be a teaching tool for people to improve their understanding of Convolutions for example, and useful to "weigh" a particular network.
This doesn't need to clutter the UI, it can totally reside in the Op/Net details panel.
The text was updated successfully, but these errors were encountered: