网络训练IndexError: Target 1 is out of bounds.问题的解决参考
网络训练IndexError: Target 1 is out of bounds.问题的解决参考问题出现在训练用的标签处，标签“1”超过范围，我这里标签是0和1，意味着原本我标签设置的范围只有一个，给了0.所以要提升维度为2.经过查找我是全连接哪里出了问题：Linear层输出维度为一，因此改为我们对应要的标签数量（2），成功。希望能帮到你...
·
网络训练IndexError: Target 1 is out of bounds.问题的解决参考
问题出现在训练用的标签处，标签“1”超过范围，我这里标签是0和1，意味着原本我标签设置的范围只有一个，给了0.所以要提升维度为2.总而言之是目标分类数量超过了你所设置的标签数量的范围。经过查找我是全连接哪里出了问题：
Linear层输出维度为一，因此将维度改为我们对应要的标签数量：2，可以通过调整维数（通俗的说，列数）解决问题，成功。希望能帮到你
有“AI”的1024 = 2048，欢迎大家加入2048 AI社区
更多推荐
33
0- 0
已为社区贡献1条内容
所有评论(0)