栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Python

CUDA error: device-side assert triggered

Python 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

CUDA error: device-side assert triggered

CUDA error: device-side assert triggered 触发了设备端断言
  • 原因1:模型大小不匹配

  • 在定义模型的最终全连接层时,我没有将 196(斯坦福汽车数据集的类总数)作为输出单元的数量,而是使用了 195。错误通常在您执行反向传播的行中识别。您的损失函数将比较模型的输出和数据集中该观察的标签。万一您对标签和输出感到困惑,请参阅下面我如何定义它们:

  • 原因2:损失函数输入错误

  • 损失函数对于它们可以接受的可能输入具有不同的范围。如果您为输出层选择不兼容的激活函数,则会触发此错误。例如,BCELoss 要求其输入介于 0 和 1 之间。如果输入(模型的输出)超出该特定损失函数的可接受范围,则会触发错误。

后来尝试将F.mse_loss 改为nn.MSELoss
  • Bool value of Tensor with more than one value is ambiguous in Pytorch
    Bool value of Tensor with more than one value is ambiguous当您尝试将张量转换为布尔值时会出现错误消息。

  • 原来是没有实例化

  • self.loss = nn.MSELoss() 在forwrad时 theloss = self.loss(x, y)

一步步调试发现Tensor显示 Unable to get repr for
  • 是因为写的程序的bug,一步步运行,仔细检查变量发现变量用错了。
参考

添加链接描述

转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/1012390.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号