Yes Sir, there is something wrong with my data but it is normal tensor data, Can you tell what can be wrong in it. I had to set requires grad True for loss as the model was returning it with requires grad False.
Thanks for quick reply.