pytorch模型在计算loss_fn(output, y_train)时，遇到RuntimeError: CUDA error: device-side assert triggered
将 x = self.classification(x)修改成 x = F.softmax(x, dim=1)。解决措施：在找了好久的原因之后，确定输入的形状没有问题，最终是output 的形状大小有问题。
·
错误情况如下图
loss_fn(output, y_train)
fn=CrossEntropyLoss()
解决措施：在找了好久的原因之后，确定输入的形状没有问题，最终是output 的形状大小有问题
具体做法：
将 x = self.classification(x)修改成 x = F.softmax(x, dim=1)。
最终实现正常运行
DAMO开发者矩阵，由阿里巴巴达摩院和中国互联网协会联合发起，致力于探讨最前沿的技术趋势与应用成果，搭建高质量的交流与分享平台，推动技术创新与产业应用链接，围绕“人工智能与新型计算”构建开放共享的开发者生态。
更多推荐
8
0- 0
已为社区贡献2条内容
所有评论(0)