I found your formulate your label as 1 and -1, and how to distinguish the label since MSELoss calculate the loss but dismiss the sign bit?