yes,you are right! @ptrblck
I remove softmax and use raw logits to the criterion, train sets and val sets are more accuracies than before, train acc: 0.96 ,val acc:0.99.  No add weighted  ,test acc:0.26, after add weighted ,test acc:0.29(slightly increased ?)