Hi @ptrblck ,
Not in the original notebook. I’d incorporated wandb for experiment tracking in mine.
A similar question was raised here in the forum (RuntimeError: Module backward hook for grad_input is called before the grad_output one. This happens because the gradient in your nn.M odule flows to the Module's input without passing through the Module's output - #3 by morg 38) a few days back.
I had replied to that post to check whether the issue was resolved. Looks like wandb.watch was the reason and on removing that, it went through is what @GeoffNN had replied. I tried the same in mine and it went through. Though I’m not sure what’s the connection between those.
Would be curious to understand that. Thanks for checking.