Hi Amiya, as I said on my previous post the problem turn out to be on my model itself. So if you already use model.eval() before doing inference with your trained model then that means there is a possibility that your model has overfitted. At least that was what happened on mine