Why doesn't want my simple pytorch network move on Cuda?
Clash Royale CLAN TAG #URR8PPP Why doesn't want my simple pytorch network move on Cuda? I built a simple network from a tutorial and I got this error: RuntimeError: Expected object of type torch.cuda.FloatTensor but found type torch.FloatTensor for argument #4 'mat1' Any help? Thank you! import torch import torchvision device = torch.device("cuda:0") root = '.data/' dataset = torchvision.datasets.MNIST(root, transform=torchvision.transforms.ToTensor(), download=True) dataloader = torch.utils.data.DataLoader(dataset, batch_size=4) class Net(torch.nn.Module): def __init__(self): super(Net, self).__init__() self.out = torch.nn.Linear(28*28, 10) def forward(self, x): x = x.view(x.size(0), -1) x = self.out(x) return x net = Net() net.to(device) for i, (inputs, labels) in enumerate(dataloader): inputs.to(device) out = net(inputs) 1 Answer 1 ...