../aten/src/ATen/native/cuda/IndexKernel.cu:92: operator(): block: [651,0,0], thread: [127,0,0] Asse

问题

原因

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

Linux上统计文件夹下文件个数以及目录个数

数据处理过程

实时查看nvidia显卡显存

watch -n 1 nvidia-smi 

#其中，1表示每隔1（可改）秒刷新一次终端的显示结果