Hello, crossentropyloss is a good place to start: CrossEntropyLoss — PyTorch 1.8.1 documentation
Depending on the level of noise in your dataset however you may need to utilize a more customized loss function