Closed hfarhid closed 3 years ago
Hey,
I am trying to run the code on customized number of classes (less than 24), but getting cuda indexing error and found the source of the error here
https://github.com/wei-tim/YOWO/blob/94c554efecb7354bc03f53818cc8cb51675db9e8/region_loss.py#L258
because num_classes is hardcoded to 24. After fixing it that still getting same error. Could give me hint how I can train model with different number of classes?
PS: cuda error : CUDA error: device-side assert triggered
Hey,
I am trying to run the code on customized number of classes (less than 24), but getting cuda indexing error and found the source of the error here
https://github.com/wei-tim/YOWO/blob/94c554efecb7354bc03f53818cc8cb51675db9e8/region_loss.py#L258
because num_classes is hardcoded to 24. After fixing it that still getting same error. Could give me hint how I can train model with different number of classes?
PS: cuda error : CUDA error: device-side assert triggered