我正在尝试 在我的自定义形状数据集上从此 repo https://github.com/eriklindernoren/PyTorch-YOLOv3训练 yolo-v3 模型，但我不断收到错误“运行时错误：CUDA 错误：触发设备端断言”
我试图查找解决方案并尝试了不同答案中建议的几件事（例如修复注释中类的索引），但错误仍然存在。
我正在按照 repo 自述文件中的描述对自定义数据集进行训练，并相应地调整了 custom.data 和 data/custom/。
我一直收到这个输出。
C:/w/1/s/windows/pytorch/aten/src/ATen/native/cuda/IndexKernel.cu:60: block: [0,0,0], thread: [32,0,0] Assertion `index >= -sizes[i] && index < sizes[i] && "index out of bounds"` failed.
C:/w/1/s/windows/pytorch/aten/src/ATen/native/cuda/IndexKernel.cu:60: block: [0,0,0], thread: [33,0,0] Assertion `index >= -sizes[i] && index < sizes[i] && "index out of bounds"` failed.
C:/w/1/s/windows/pytorch/aten/src/ATen/native/cuda/IndexKernel.cu:60: block: [0,0,0], thread: [34,0,0] Assertion `index >= -sizes[i] && index < sizes[i] && "index out of bounds"` failed.
C:/w/1/s/windows/pytorch/aten/src/ATen/native/cuda/IndexKernel.cu:60: block: [0,0,0], thread: [35,0,0] Assertion `index …