1. 数据输入
pytorch 中计计算交叉熵损失函数时, 输入的正确 label 不能是 one-hot 格式。函数内部会自己处理成 one hot 格式。所以不需要输入 [ 0 0 0 0 1],只需要输入 4 就行。
在经过 loss 的时候,CrossEntropyLoss 会自动为其编码为 one-hot 编码
CrossEntropyLoss
2. target取值范围
Target为100,则是从0~99