-
Notifications
You must be signed in to change notification settings - Fork 281
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
Add variable names to nodes #59
Comments
Hi, |
Completely untested:
|
The problem is that most of the operations happen in c++ and the python |
Hmm. How about something like this, then? Even more untested, if that's even possible.
Usage: |
In support of this, adding variable names to nodes would be helpful to annotate exactly which tensors in the code are being saved for backwards |
This is neat and by solving an NP-hard problem in my head I can tell which node corresponds to a given other tensor in my code. Could you add the names of each tensor to its node in the graph? Perhaps using https://pypi.org/project/varname/ or by modifying
Tensor.__init__
to extract variable names from declarations in the traceback, then saving them ingrad_fn
.The text was updated successfully, but these errors were encountered: