Error in the forecast

[dafmdev]

Thank you so much for this great work!

Using the model in ckpt the following error appears:

RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cpu and cuda:0! (when checking argument for argument index in method wrapper_CUDA__index_select)

It is possible that it is because of the .to('cuda').

[tangh18]

It may raise this error since this repo is built mainly for my own research purpose where I assume all the operations is running on a cuda device. I believe it's convenient to adjust the codes in this repo to fit other devices like cpu and the correctness remains.