Skip to content

Week0 run: Fixed error with usage of GPU (or CPU)#41

Open
KansaiUser wants to merge 4 commits intograviraja:mainfrom
KansaiUser:week0_run
Open

Week0 run: Fixed error with usage of GPU (or CPU)#41
KansaiUser wants to merge 4 commits intograviraja:mainfrom
KansaiUser:week0_run

Conversation

@KansaiUser
Copy link
Copy Markdown

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.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant