Comment
Author: Admin | 2025-04-28
I am trying to train the yolo-v3 model from this repo https://github.com/eriklindernoren/PyTorch-YOLOv3on my custom dataset of shapes, but I keep getting the error "RuntimeError: CUDA error: device-side assert triggered"I have tried to lookup the solution and tried several things suggested in different answers (like fixing the indexing of the classes in the annotations) but the error persists.I am following the description in the readme of the repo to train on a custom dataset, and have adjusted custom.data and the data/custom/ accordingly.I keep receiving this output.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 = -sizes[i] && index = -sizes[i] && index = -sizes[i] && index = -sizes[i] && index = -sizes[i] && index = -sizes[i] && index = -sizes[i] && index = -sizes[i] && index = -sizes[i] && index = -sizes[i] && index = -sizes[i] && index = -sizes[i] && index = -sizes[i] && index = -sizes[i] && index = -sizes[i] && index = -sizes[i] && index = -sizes[i] && index = -sizes[i] && index = -sizes[i] && index = -sizes[i] && index = -sizes[i] && index = -sizes[i] && index = -sizes[i] && index = -sizes[i] && index = -sizes[i] && index = -sizes[i] && index = -sizes[i] && index = -sizes[i] && index = -sizes[i] && index = -sizes[i] && index = -sizes[i] && index = -sizes[i] && index = -sizes[i] && index = -sizes[i] && index = -sizes[i] && index loss, outputs = model(imgs, targets) File "C:\Users\user\AppData\Local\Programs\Python\Python36\lib\site-packages\torch\nn\modules\module.py", line 547, in __call__
Add Comment