custom backward function

Open vishwali opened this issue 8 years ago • 0 comments

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

Jan 20 '18 21:01 vishwali