huihuiM:

[-0.9032, -0.5197]


yes, the problem  is the binary classification. But I have not do the one hot encoding for the output model.
the last operation of the model is:
F.log_softmax(pred_model)
Is there some problem?