【pytorch】thread: [16,0,0] Assertion `t >= 0 && t < n_classes` failed

**报错信息:**C:\cb\pytorch_1000000000000\work\aten\src\ATen\native\cuda\Loss.cu:250: block: [0,0,0], thread: [16,0,0] Assertion `t >= 0 && t < n_classes` failed

**可能原因:**标签label或者预测的结果pred,超出了数据的范围,比如标签里面只有0-15的数字,但是pred中出现19这个数字,那么在交叉熵损失计算中就会报错。

相关推荐
AI蜗牛之家1 小时前
Qwen系列之Qwen3解读:最强开源模型的细节拆解
人工智能·python
王上上1 小时前
【论文阅读30】Bi-LSTM(2024)
论文阅读·人工智能·lstm
whyeekkk2 小时前
python打卡第48天
开发语言·python
YunTM2 小时前
贝叶斯优化+LSTM+时序预测=Nature子刊!
人工智能·机器学习
舒一笑3 小时前
智能体革命:企业如何构建自主决策的AI代理?
人工智能
SpikeKing4 小时前
Server - 使用 Docker 配置 PyTorch 研发环境
pytorch·docker·llm
丁先生qaq4 小时前
热成像实例分割电力设备数据集(3类,838张)
人工智能·计算机视觉·目标跟踪·数据集
Eiceblue4 小时前
Python读取PDF:文本、图片与文档属性
数据库·python·pdf
红衣小蛇妖4 小时前
神经网络-Day45
人工智能·深度学习·神经网络
weixin_527550404 小时前
初级程序员入门指南
javascript·python·算法