Am I training the model correctly?

4

This question was really helpful for my question later one.

Thank you @ptrblck

I used softmax function two times.

  1. output of model
  2. put the softmax output to the nn.CrossEntropyLoss()