Hi , how did you solved problem? would you please explain whole custom loss how can be calculate with nice example.
Alban said  we have to compute derivative of  loss wrt input (dl/dinp) when using autograd but in tutorials custom relu function  why we didnt calculated grad_input[input > 0 ] = 1.please would you give example on mse loss as you solved . Thanks