Hi,
After defining a custom layer with customized gradient computation, do I need to have the gradients with requires_grad=True? Or is it enough just to have the requires_grad as true for the input and the output of that layer?
Does it make a problem if unnecessarily I put requires_grad as True for some variables?
Thanks in advance!