Right! Sorry about that.
So my criterion is defined right before running my training with this code:
criterion = nn.CrossEntropyLoss()
I can choose another loss function other than CrossEntropyLoss? Which would you recommend?