Sowmya_Krishnan:

One question though - I tried disabling cudnn just before I call the backward() function. Is that a reason for it not working? Should I disable it at the point where the training process begins?


Yes, that’s the reason, as you should disable it for the forward and backward pass for this layer.