@ptrblck hmm, the thing is I’m not explicitly registering a backward hook anywhere in my module.
My module consists of a bunch of normal PyTorch modules/functions which have their own backward methods. So I don’t see the need to write my own backward method. Do I need to write one now?