search

shobrook / sequitur

Library of autoencoders for sequential data
MIT License
410 stars 56 forks source link

How to use cuda to accelerate when using quick_train #12

Open midiexianghhh opened 1 year ago

midiexianghhh commented 1 year ago

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

SecantZhang commented 3 months ago

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