Greetings. I have just started checking your repo, and I liked the week 0 content.
I edited both the training and inference scripts in order to make it able to correctly use the GPU.
(and solve the [RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cpu! )
Other modifications are minor to explain better the notebook or readme.
Greetings. I have just started checking your repo, and I liked the week 0 content.
I edited both the training and inference scripts in order to make it able to correctly use the GPU. (and solve the [RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cpu! )
Other modifications are minor to explain better the notebook or readme.