medium_articles icon indicating copy to clipboard operation
medium_articles copied to clipboard

Published 20 hours ago • verified publisher icon erykml

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

Open Lootwig opened this issue 3 years ago • 0 comments

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 avatar Dec 14 '20 23:12 Lootwig