search

nok-halfspace / Transformer-Time-Series-Forecasting

373 stars 102 forks source link

CUDA enabled? #9

Open 1-NoLimits opened 2 years ago

1-NoLimits commented 2 years ago

Hi,

I set device to cuda and default also to cuda but it runs on cpu. Further inspection shows .cpu() hardcoded across the scrips.

Is there a version that uses cuda?

jdwillard19 commented 2 years ago

If you set the device parameter equal to "cuda:0" or simply move the model and and data to the GPU (e.g. model = model.cuda(); data = data.cuda()) this runs fine on GPU as cuda-enabled.