loss_fn 是 nn.CrossEntropyLoss() 交叉熵,那么这个输出类别似乎是没有通过softmax 直接就输入交叉熵了吗? 如果要softmax+交叉熵的话,是应该在定义网络 ...
確定! 回上一頁