Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

lenet5_pytorch.ipynb: Executing code from "In [12]" crashes program #2

Open
Lootwig opened this issue Dec 14, 2020 · 0 comments
Open

Comments

@Lootwig
Copy link

Lootwig commented Dec 14, 2020

The line _, probs = model(valid_dataset[index][0].unsqueeze(0)) throws RuntimeError: Input type (torch.FloatTensor) and weight type (torch.cuda.FloatTensor) should be the same

@Lootwig Lootwig changed the title Executing code from "In [12]" crashes program lenet5_pytorch.ipynb: Executing code from "In [12]" crashes program Dec 14, 2020
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

No branches or pull requests

1 participant