device-side assert triggered at /pytorch/aten/src/THC/THCReduceAll.cuh:327
关于device-side assert triggered at /pytorch/aten/src/THC/THCReduceAll.cuh:327
主要是因为交叉熵损失函数的labels出现了越界，labels的值只能为[0,n_class-1]。
之前更改了Maskrcnn的数据集后类的个数应该为50个，但是数据处理的时候直接设置到了50所以一直报错！将label的tensor - 1就完美解决了！
如下代码中注意predicate_labels中的取值即可！
self.criterion = nn.CrossEntropyLoss()
prm_loss = self.criterion(output, predicate_labels)
device-side assert triggered at /pytorch/aten/src/THC/THCReduceAll.cuh:327相关推荐
- PyTorch报错：RuntimeError: CUDA error: device-side assert triggered at /pytorch/aten/src/THC/generic
训练模型报错 RuntimeError: cuda runtime error (710) : device-side assert triggered at /pytorch/aten/src/TH ...
- PyTorch 错误 RuntimeError: invalid argument 5: k not in range for dimension at /pytorch/aten/src/THC/g
PyTorch 错误 RuntimeError: invalid argument 5: k not in range for dimension at /pytorch/aten/src/THC/g ...
- RuntimeError: cuda runtime error (3) : initialization error at /pytorch/aten/src/THC/THCGeneral.cpp:
第三个大坑:在python多进程中使用pytorch加载模型时报错: THCudaCheck FAIL file=/pytorch/aten/src/THC/THCGeneral.cpp line=5 ...
- 深度学习报错 | THCudaCheck FAIL file=/pytorch/aten/src/THC/THCGeneral.cpp
深度学习报错 | THCudaCheck FAIL file=/pytorch/aten/src/THC/THCGeneral.cpp 错误定位 解决历程 错误定位 近日在自己的服务器上跑别人的代码时 ...
- 解决RuntimeError: cuda runtime error (30) : unknown error at /pytorch/aten/src/THC/THCGeneral.cpp:70N
解决两个问题: (1)RuntimeError: cuda runtime error (30) : unknown error at /pytorch/aten/src/THC/THCGeneral ...
- [debug]THCudaCheck FAIL file=/pytorch/aten/src/THC/THCGeneral.cpp line=405 error=11 : invalid argume
解决 THCudaCheck FAIL file=/pytorch/aten/src/THC/THCGeneral.cpp line=405 error=11 : invalid argument 完 ...
- RuntimeError: cuda runtime error (30) : unknown error at /pytorch/aten/src/THC/THCGeneral.cpp:50
RuntimeError: cuda runtime error (30) : unknown error at /pytorch/aten/src/THC/THCGeneral.cpp:50 在运行 ...
- pytorch 报错“THCudaCheck FAIL file=/pytorch/aten/src/THC/THCGeneral.cpp line=663 error=11“解决方案
今天使用pysot测试数据,pytorch出现下面报错: THCudaCheck FAIL file=/opt/conda/conda-bld/pytorch_1535493744281/work/a ...
- python出现THCudaCheck FAIL file=/pytorch/aten/src/THC/ThCGeneral.cpp line=405 error=11 : 情况
python出现THCudaCheck FAIL file=/pytorch/aten/src/THC/ThCGeneral.cpp line=405 error=11 : invalid argum ...
最新文章
- 一文读懂GAN, pix2pix, CycleGAN和pix2pixHD
- python 数据分析学什么-入门数据分析,我应该学习什么编程语言
- 关于codeblocks插件（持续更新）
- linux查usb版本,怎么查看 ubuntu usb版本
- 剑指Offer之寻找数据流中的中位数【包含大顶堆小顶堆解释】
- 技术人员如何成长？（资料整理）
- 3.3 keras模型构建的三种方式
- maven 配置tomcat 插件
- 取整函数（ceil、floor、round）
- PyQt5桌面应用开发----环境安装配置及第一个桌面应用程序
- 视频提取关键帧的三种方式【已调通】
- 可用资源不足excel无法完成任务_你若会用Excel里的SUM函数，又何须加班到半夜！【Excel教程】...
- Codeforces 1635 E. Cars 二分图+拓扑排序
- D:\anaconda3\envs\py2\python.exe: can‘t open file ‘create‘: [Errno 2] No such file or directory
- 第五章（第四节）论路由器重置
- 双十二有哪些数码好物值得入手、双十二必买数码好物清单
- 一分钟看懂Python中的 // 和 / 和 % 的用法区别
- 报告指出，中国市场巩固了作为智利车厘子全球主要出口目的国的市场地位
- Unity3d 屏幕空间人体皮肤知觉渲染次表面散射Screen-Space Perceptual Rendering Subsurface Scattering of Human Skin
- saber软件安装后怎么打开_Saber中文使用教程之软件仿真流程