问题:多分类网络加了两层全连接后最后输出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
本文解析了多分类网络中加入两层全连接层后,输出单一类别时出现的TargetN is out of bounds错误原因及解决办法。重点在于检查网络输出的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
427
396
259

被折叠的 条评论
为什么被折叠?