GPU usage

[XilunZhangRobo]

Hi thanks for the amazing codebase. I am wondering does the code use gpu during training? I did not find anything relevant on setting the device for training in the code.

[mayhorizon]

Hi,
Yes I think it does. Otherwise you can set the training with multi cores and it doesn't run that slowly (20 minutes for 10,000 epochs with 32 cores)

[SvenGronauer]

Hi,

the training code examples are implemented for CPU usage. However, with some PyTorch code adjustments you can run the code also on a GPU.

Best, Sven