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)。
最终实现正常运行
腾讯云面向开发者汇聚海量精品云计算使用和开发经验，营造开放的云计算技术生态圈。
更多推荐
8
0- 0
已为社区贡献3条内容
所有评论(0)