Hi Frank! Thanks so much for this detailed response, it really helped clarify things for me. Your assessment in reading between the lines was absolutely correct, so I think I’ll be sticking with just regular CrossEntropyLoss for now and see how that goes.