sbarratt/torch_cg

X0 undefined in backward pass

Closed this issue · 3 comments

X0 seem to be used but undefined in the backward pass of the current implementation.

Note: it might be worth adding a test with an X0.

I am not sure the backward pass works when there is a X0 as it does not return a gradient for X0 (I encountered the bug but it was with a heavily modified version of your code so it might not be relevant).