Some methods in TorchHook need docstrings. In particular, some of the original hooking methods (for example here and here and here) are missing them, as well as all the fixed precision ones (mainly all of the ones in this guy here).
This is a great issue for people who want to understand more of TorchHook, since this will require clearly understanding and explaining the purpose of each of these methods and functions.
I will take on this issue
Are we still missing some docstrings?
@kamathhrishi Any Update ?
I can check with you if you want ;)
If it's still open I will try to tackle this issue. First, I will try to create docstring for _hook_module.
I am interested in taking a stab at the issue.
Most helpful comment
This is a great issue for people who want to understand more of TorchHook, since this will require clearly understanding and explaining the purpose of each of these methods and functions.