How to use cuda to accelerate when using quick_train

[midiexianghhh]

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

[SecantZhang]

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