jcjohnson / jcjohnson/pytorch-examples
custom backward function
Open
- Dominant language
- Python
- Stars
- 4.9k
- Forks
- 918
- PR merge metrics
- No merged PRs in 30d
Description
https://github.com/jcjohnson/pytorch-examples/blob/0f1b88a38e1fd761dd4ee22398d0efec7a3c7faf/autograd/two_layer_net_custom_function.py#L30
self.saved_tensors in custom backward function doesn't work hence cannot compute backward using the subclass method
Contributor guide
No contributing guide indexed for this repository
Assessment
This issue has not been assessed yet.