Doesn't work with CUDA mode

ServiceNow / N-BEATS

N-BEATS is a neural-network based model for univariate timeseries forecasting. N-BEATS is a ServiceNow Research project that was started at Element AI.

Other

508 stars 116 forks source link

Doesn't work with CUDA mode #18

Open sachinjm opened 1 year ago

sachinjm commented 1 year ago

Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cpu!