Hi, I find the code uses F.log_softmax() and nn.nLLLoss() to count the softmax entropy loss. And I use nn.CrossEntropyLoss() instead.
確定! 回上一頁