Open Rajat-Mehta opened 4 years ago
The error is in this line, File "/SA-SSD/mmdet/models/single_stage_heads/ssd_rotate_head.py", line 367, in get_guided_anchors:
top_labels = torch.cat([gt_lbls, top_labels],0)
For now I have fixed this issue by adding this line:
top_labels=top_labels.to(gt_lbls.device) // i.e. gpu
top_labels = torch.cat([gt_lbls, top_labels],0)
is there a better solution for this ?
The error is in this line, File "/SA-SSD/mmdet/models/single_stage_heads/ssd_rotate_head.py", line 367, in get_guided_anchors:
top_labels = torch.cat([gt_lbls, top_labels],0)
For now I have fixed this issue by adding this line:
top_labels=top_labels.to(gt_lbls.device) // i.e. gpu top_labels = torch.cat([gt_lbls, top_labels],0)
is there a better solution for this ?
THX! I have met the same problem, and have been solved by your solution!
I am not able to train the model, getting error related to CUDA.
Any idea why am I getting this error and how to solve it ?
Also, I am getting a lot of warnings related to numba, how to resolve them?
Thanks in advance