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.

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

Added cuda checks to enable inference via the universal demo #7

Open
wants to merge 1 commit into
base: master
Choose a base branch
from
Open

Added cuda checks to enable inference via the universal demo #7

wants to merge 1 commit into from

Conversation

Jrokisky
Copy link

@Jrokisky Jrokisky commented Aug 4, 2020

Hi all,
I added some conditionals around the cuda calls to enable some inference via cpu.
The changes made allowed me to successfully run:

python -u mseg_semantic/tool/universal_demo.py   --config=mseg_semantic/config/test/default_config_360_ss.yaml   model_name mseg-3m   model_path ./mseg-3m-480p.pth input_file $IMAGE

@johnwlambert
Copy link
Collaborator

thanks very much for this contribution @Jrokisky , I've been meaning to add the cpu inference support. I'm traveling the next few days but will double check the integration tests when I get back and try to get this merged in by early next week

@Jrokisky
Copy link
Author

Jrokisky commented Aug 4, 2020

@johnwlambert Sounds good. Thanks for sharing your work in such an easy to use way!

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants