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

colab cpu inference #1

Open

AK391 opened this issue Jul 14, 2021 · 3 comments

AK391 commented Jul 14, 2021

great work, is it possible to do inference on cpu in colab?

Owner

mchong6 commented Jul 16, 2021

Added device which lets you use cpu. But runs out of memory on colab if doing too many images so have to be careful for that.

Author

AK391 commented Jul 16, 2021

@mchong6 thanks the projection part still requires gpu?

Owner

mchong6 commented Jul 17, 2021

Sorry, just fixed that!

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