torch的交叉熵函数坑点
nn.CrossEntropyLoss(pred,label)
该模块提供的交叉熵函数
第一个参数注意点
该函数已经帮我们做了softmax操作和log操作
所以针对预测值pred我们只要输入对应的原始值
即我们只需要输入对应的线性层输入,并不需要激活,否则精度会大大下降。
第二个参数注意点
1、 label标签值 我们不需要做one-hot编码 输入原始编号即可
2、label只支持一维tensor 也就是说一次batch-size训练中有[batch-size,1]数据 我们需要把label变为[batch-size]的一维label
3、第二个参数只能接收long型数据