nn.MSELoss and a softmax output look a bit strange. Could you describe your use case a bit more, please?
I assume the face expression recognition project is a multi-class classification, so I would expect to see nn.CrossEntropyLoss as the criterion.