Erro no código
tensor = torch.from_numpy(X1).float() # X1 é um array numpy. ".float()" faz a conversão para float32
pred = net(tensor)
print(pred.size()) # torch.Size([300, 1]) - 300 predições com 1 dimensão
Erro:
RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cpu! (when checking argument for argument mat1 in method wrapper_CUDA_addmm)
Adicionando:
...
net = net.to('cpu')
pred = net(tensor)
...
Update
Consegui executar no cuda usando torch.device
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
tensor = tensor.to(device)
net = net.to(device)
Fica para referência futura
