码上快乐

相关内容   简体   繁体

Pytorch: cuda runtime error (59) : device-side assert triggered at /pytorch/aten/src/THC/generic/THCTensorMa

本文转载自  查看原文  2019-04-04 10:49  10561    深度学习/ 杂七杂八


更换了数据集, 在计算交叉熵损失时出现错误 :
cuda runtime error (59) : device-side assert triggered at /pytorch/aten/src/THC/generic/THCTensorMa

解决检查两个问题:

1. 模型输出label数量 是否与 标签类别数量相同

<label为 男,女, 未知;  但是我模型设计只有男/女, 也就是label的标记数字超出了模型分类数>

2. label是否有-1的情况,需要提前过滤掉,类似下面这样的标签


免责声明！

本站转载的文章为个人学习借鉴使用，本站对版权不负任何法律责任。如果侵犯了您的隐私权益，请联系本站邮箱yoyou2525@163.com删除。



猜您在找 Pytorch报错：cuda runtime error (59) : device-side assert triggered at /pytorch/aten/src/THC/generic/THCTensorMath.cu:26 RuntimeError: cuda runtime error (710) : device-side assert triggered at /pytorch/aten/src/THC/generic/THCTensorMath.cu:26 RuntimeError: cuda runtime error (59) : device-side assert triggered（已解决） CUDA error: device-side assert triggered RuntimeError: CUDA error: device-side assert triggered的解决 完美解决-RuntimeError: CUDA error: device-side assert triggered 解决RuntimeError: cuda runtime error (30) : unknown error at /pytorch/aten/src/THC/THCGeneral.cpp:70&NVIDIA-SMI has failed because it couldn't communicate with the NVIDIA driver. Make sure that the late RuntimeError: size mismatch, m1: [32 x 2304], m2: [400 x 100] at C:/w/1/s/windows/pytorch/aten/src\THC/generic/THCTensorMathBlas.cu:268 Pytorch的runtime error pytorch 使用gpu报错CUDA error: no kernel image is available for execution on the device
 
粤ICP备18138465号  © 2018-2025 CODEPRJ.COM