nn.CrossEntropyLoss() 时报错

1. 数据输入

pytorch 中计计算交叉熵损失函数时, 输入的正确 label 不能是 one-hot 格式。函数内部会自己处理成 one hot 格式。所以不需要输入 [ 0 0 0 0 1],只需要输入 4 就行。

在经过 loss 的时候,CrossEntropyLoss 会自动为其编码为 one-hot 编码

2. target取值范围

 Target为100,则是从0~99

 

posted @ 2022-11-10 14:45  wangssd  阅读(264)  评论(0编辑  收藏  举报