我正在运行一个简单的神经网络,其中包含大约 1112 行、23 个输入、2 个隐藏层和 31 个可能输出的 csv 数据。前向训练后,在执行以下代码期间,我收到以下错误消息
该行执行错误----->loss = criteria(y_pred, y_train)
IndexError Traceback(最近一次调用最后一次)
在
3帧 /usr/local/lib/python3.10/dist-packages/torch/nn/function.py in cross_entropy(输入、目标、权重、size_average、ignore_index、reduce、reduction、label_smoothing) 第3084章 第3085章 - > 3086返回torch._C._nn.cross_entropy_loss(输入,目标,权重,_Reduction.get_enum(减少),ignore_index,label_smoothing) 3087 3088
IndexError:目标 32 超出范围。
如有任何建议,我们将不胜感激。提前致谢。最好的问候。
尝试前向训练并使用 loss = criteria(y_pred, y_train) 计算损失
您有
32
课程(编号从 0
到 31
)。
错误
IndexError: Target 32 is out of bounds.
表示
32
数组中有一个 target
,而值应该是类,因此它们的范围应从 0
到 31
。
检查您的数据生成?