sequitur icon indicating copy to clipboard operation
sequitur copied to clipboard

Published 20 hours ago • verified publisher icon shobrook

How to use cuda to accelerate when using quick_train

Open midiexianghhh opened this issue 1 year ago • 1 comments

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

midiexianghhh avatar Apr 28 '23 07:04 midiexianghhh

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

SecantZhang avatar Apr 03 '24 03:04 SecantZhang