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

How to use cuda to accelerate when using quick_train #12

Open

midiexianghhh opened this issue Apr 28, 2023 · 1 comment

midiexianghhh commented Apr 28, 2023 •

edited

How to use cuda to accelerate when using quick_train, does the method quick_train provide some parameter to specify the device

midiexianghhh changed the title ~~How~~ How to use cuda to accelerate when using quick_train

SecantZhang commented Apr 3, 2024

I know it might be a bit outdated but yes, there's a parameter called device now.
train_model(..., device=torch.device("cuda"))

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