1. neg_log_likelihood = F.nll_loss(Y_prob, Y, weight=weight)
input, target 位置不要弄反,多类nll loss可以把实数编码转为one hot编码,此处不需要转换
2. torch.LongTensor(Y), 数据类型
此处两个错误自己易犯的,不容易察觉
1. neg_log_likelihood = F.nll_loss(Y_prob, Y, weight=weight)
input, target 位置不要弄反,多类nll loss可以把实数编码转为one hot编码,此处不需要转换
2. torch.LongTensor(Y), 数据类型
此处两个错误自己易犯的,不容易察觉