-
Notifications
You must be signed in to change notification settings - Fork 9.5k
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Fix device of the tensors in set_nms #10574
Conversation
@max-unfinity Hi. please fix the lint |
@hhaAndroid HI! |
Hi @max-unfinity, We'd like to express our appreciation for your valuable contributions to the mmdetection. Your efforts have significantly aided in enhancing the project's quality. If you're on WeChat, we'd also love for you to join our community there. Just add our assistant using the WeChat ID: openmmlabwx. When sending the friend request, remember to include the remark "mmsig + Github ID". Thanks again for your awesome contribution, and we're excited to have you as part of our community! |
hi,have solved the problem? could you tell me how to do? thank you |
Motivation
Fixed
RuntimeError: indices should be either on cpu or on the same device as the indexed tensor (cpu)
when running CrowdDet model on cuda device (the model uses MultiInstanceBBoxHead roi_head)Modification
Just added
tensor.to(device)
to fix the issue inset_nms
method.versions: