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

PyTorch错误定位系列之CUDA error: device-side assert triggered

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

PyTorch错误定位系列之CUDA error: device-side assert triggered

PyTorch错误定位系列之CUDA error: device-side assert triggered Introduction

本栏目只是提供一些自己遇到的错误的解决思路。

Background

我昨天写了个模型加了focal loss可以训练,今天换了一批数据,尼玛第二个epoch就报了上面标题这个错误。

Solution

经过本人,本菜鸡的大量调研(google),发现该错误可以总结为tensor溢出了或者下溢了。

所以按着这个思路,我就尝试把我加的focal loss给去掉了,换成了正常的nn.CrossEntropyLoss 就尼玛能正常了。

因此,我猜测这回真的遇到了数值溢出。因为在我抄过来的focal loss中有一个torch.exp的操作,由于写得时间过早(2周前),我想不起来是原作者就这样还是我改的有问题。 最终,我就的方法就是用nn.CrossEntropyLoss,嘿嘿嘿。

开玩笑哈,知道问题,我们就知道往哪里改了。

改,我肯定要改的!!

Reference
  1. CUDA error: device-side assert triggered解决方案
  2. PyTorch 错误 RuntimeError: CUDA error: device-side assert triggered
转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/315786.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

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

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