Solved by KFrank in post #2 
              
              
                Hi Had! 

The short answer is that MSE loss is not a loss that would naturally 
be used with classes.  As such, pytorch’s MSELoss does not attempt 
to implement any feature that might be considered analogous to 
“class weights.” 
The reason is that MSE loss is naturally a measure how much two 
sets …