`get_detector_inference_runner`: resolve the device from the model config when None by nattse · Pull Request #2841 · DeepLabCut/DeepLabCut

Skip to content

Navigation Menu

Sign in

Appearance settings

Provide feedback

We read every piece of feedback, and take your input very seriously.

Saved searches

Use saved searches to filter your results more quickly

Sign up

Appearance settings

Conversation

@nattse

Copy link

Contributor

@nattse nattse commented

Jan 20, 2025

Fixes #2840 where get_detector_inference_runner was not able to get device from model_cfg, leaving all detector inference to be done on CPU

Copy link

Contributor

@n-poulsen n-poulsen left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thanks @nattse!

@n-poulsen n-poulsen merged commit db23d67 into DeepLabCut:pytorch_dlc

Jan 21, 2025

@n-poulsen n-poulsen changed the title Update utils.py get_detector_inference_runner: resolve the device from the model config when None

Jan 21, 2025

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Reviewers

1 more reviewer

@n-poulsen n-poulsen n-poulsen approved these changes

Reviewers whose approvals may not affect merge requirements

Assignees

No one assigned

Projects

None yet

Milestone

No milestone

Development

Successfully merging this pull request may close these issues.

2 participants

@nattse @n-poulsen