@sai_tharun I did it, but didn’t help. It gets worse, zero accuracy for validation.
Do I need to get outputs = F.softmax(outputs, dim=1) the place I am doing or should be after last layer in the model? am I calculating accuracy correctly?