As I understood it, logits are the raw predictions that are fed into the softmax function which produces the probabilities for each class - if so, why is the final output of the model logits?
When you say F.log_softmax is calculated internally, is that done automatically?
I would really appreciate it if you could give me a working example, following the docs 1 for a classifier, they did calculate the criterion & loss in the same fashion as I did while also using cross-entropy.
Initially, I expected model.train() to train the model, but given that it is done manually I’m having difficulties understanding the process.
Edit: I lowered my learning rate.