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

[bugfix] fix predict method so it can predict use flexible device #16

Open

nurgoni opened this issue May 21, 2024 · 0 comments

Collaborator

nurgoni commented May 21, 2024

Currently, predict method by default use cuda. it is not flexible because can not used on device without cuda.

nurgoni added a commit that referenced this issue


          fix: add device parameter and locate model and data to device #16

8fbe6bc

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