Solved by KFrank in post #2 
              
              
                Hi Abhilash! 

The most straightforward approach would be to use cross entropy as 
your loss function, but to use probabilistic ("soft’) labels, rather than 
categorical (“hard”) labels. 
Pytorch does not have a soft-label version of cross-entropy loss built 
in, but it is easy to implement one.  Se…