search

wei-tim / YOWO

You Only Watch Once: A Unified CNN Architecture for Real-Time Spatiotemporal Action Localization
Other
846 stars 158 forks source link

CUDA error: device-side assert triggered #67

Closed hfarhid closed 3 years ago

hfarhid commented 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