../aten/src/ATen/native/cuda/IndexKernel.cu:92: operator(): block: [651,0,0], thread: [127,0,0] Asse
../aten/src/ATen/native/cuda/IndexKernel.cu:92: operator(): block: [651,0,0], thread: [51,0,0] Assertion `-sizes[i]
·
问题
原因
Assertion `index >= -sizes[i] && index < sizes[i] && “index out of bounds”
数据超出了边界。在给出label的时候,我的数据集在某个标签上会给进去一个小于零或者大于类别数的一个标签,大白话就是设定了模型分类数量为4,但是数据集中有大于4个类别则会报错,少于4个是不会报错。
解决方法:修改预设的类别数量,class_map里面预先写好的类别个数和实际的数据集里面的标签种类和个数不同。可以要修改class_map或者修改数据集。
这里有96的类别
但是,代码中最高95.。。
是之前02标签转换的时候类别错了。。。
参考1
解决方案
删掉02的图像和标签
rm -r 00000061*
显示文件夹下jpg图像数量
ls -lR | grep ".*jpg" | wc -l
实时查看nvidia显卡显存
watch -n 1 nvidia-smi
#其中,1表示每隔1(可改)秒刷新一次终端的显示结果
欢迎来到由智源人工智能研究院发起的Triton中文社区,这里是一个汇聚了AI开发者、数据科学家、机器学习爱好者以及业界专家的活力平台。我们致力于成为业内领先的Triton技术交流与应用分享的殿堂,为推动人工智能技术的普及与深化应用贡献力量。
更多推荐
已为社区贡献1条内容
所有评论(0)