pytorch----Target 2 is out of bounds_target 20 is out of bounds.

作者：你好赵伟 | 2024-02-27 03:22:32

踩

target 20 is out of bounds.

问题：多分类网络加了两层全连接后最后输出1类，计算loss时报错Target N is out of bounds

其中的N是正在处理的数据输入的标签，即第几类，是一个 [0, 类数-1] 的整数，最后输出1类与输入的target不符就会报错

解决方法：查看网络的最后输出，softmax的输出节点数是否等于所有的标签数。

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/你好赵伟/article/detail/149645