pytorch----Target 2 is out of bounds
问题：多分类网络加了两层全连接后最后输出1类，计算loss时报错Target Nis out of bounds其中的N其实就是处理的数据输入的标签，即第几类，是一个代表类别的整数，最后输出1类与输入的target不符就会报错解决方法：查看网络的最后输出，softmax的输出节点数是否等于所有的标签数。...
·
问题：多分类网络加了两层全连接后最后输出1类，计算loss时报错Target N is out of bounds
其中的N是正在处理的数据输入的标签，即第几类，是一个 [0, 类数-1] 的整数，最后输出1类与输入的target不符就会报错
解决方法：查看网络的最后输出，softmax的输出节点数是否等于所有的标签数。
参考：https://discuss.pytorch.org/t/indexerror-target-2-is-out-of-bounds/69614/13
汇聚全球AI编程工具，助力开发者即刻编程。
更多推荐
10
0- 0
已为社区贡献3条内容
所有评论(0)