Hi. Thank you for your suggestion. I printed the type of criterion and it is crossentropyloss. But I define it as MSELoss. Previously I used to defined it as CrossEntropyLoss but now I change it. It is weird. Could you please tell me how to fix this problem?