Skip to content
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.

Sign up for GitHub

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

Jump to bottom

Model device management after training #176

Open
gumityolcu opened this issue Sep 29, 2024 · 1 comment
Open

Model device management after training #176

gumityolcu opened this issue Sep 29, 2024 · 1 comment
Labels
bug Something isn't working

Comments

@gumityolcu
Copy link
Collaborator

Currently, SubclassDetection benchmark will have self.group_model in "cpu" even though the original model is supplied in cuda. This raises and error during the evaluate call.

How to best manage model devices after training? Is this an issue just for SubclassDetection?
@gumityolcu gumityolcu added the pre-release Before-release TODOs label Sep 29, 2024
@gumityolcu gumityolcu assigned gumityolcu and unassigned gumityolcu Sep 29, 2024
@gumityolcu gumityolcu mentioned this issue Sep 30, 2024
Merged
@gumityolcu
Copy link
Collaborator Author

This is currently handled by sending model to the device every time after training. Hence, i am removing the pre-release label

@gumityolcu gumityolcu removed the pre-release Before-release TODOs label Oct 3, 2024
@dilyabareeva dilyabareeva added the bug Something isn't working label Oct 6, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
bug Something isn't working
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
2 participants
@gumityolcu @dilyabareeva