【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这个数字,那么在交叉熵损失计算中就会报错。

相关推荐
椰皮糖10 分钟前
13.DataLoader 的使用
pytorch·深度学习·pycharm
菜菜僵尸15 分钟前
Yolo环境搭建(深度学习基础环境)
人工智能·python·深度学习
学编程的小程15 分钟前
Windows电脑本地安装跨平台文生音乐AI应用MusicGPT详细教程
人工智能·windows
西农小陈18 分钟前
python-带空格的数字层三角形
开发语言·python·算法
椰皮糖33 分钟前
16.神经网络 - 卷积层
人工智能·深度学习·神经网络
yyds(永远单身?)35 分钟前
ImportError: DLL load failed while importing _ssl: 找不到指定的模块。
python·ssl
CS_木成河38 分钟前
【HuggingFace Transformers】LlamaDecoderLayer源码解析
人工智能·pytorch·深度学习·语言模型·transformer·llama
穷人小水滴44 分钟前
(章节 3.1) 本地运行 AI 有多慢 ? 大模型推理测速 (llama.cpp, Intel GPU A770)
人工智能·llama·llama.cpp·a770·vulkan·gnu/linux·sycl
职场人参1 小时前
照片怎么压缩变小?教你8种简单好用的压缩照片方法!
人工智能·计算机视觉·音视频
爬行程序猿1 小时前
深度学习基础--深度学习网络
人工智能·深度学习·php