jcjohnson / jcjohnson/pytorch-examples

Example doesn't run on gpu (when uncommenting)

Open
#21 0 comments 0 reactions 0 assignees View on GitHub
Dominant language
Python
Stars
4.9k
Forks
918
PR merge metrics
No merged PRs in 30d

Description

`# dtype = torch.device("cuda:0") # Uncomment this to run on GPU`
should be
`# device = torch.device("cuda:0") # Uncomment this to run on GPU`

Contributor guide

No contributing guide indexed for this repository

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.