Solved by thecho7 in post #8 
              
              
                MSE takes inputs of float type. 
data = data.unsqueeze(1).type(torch.float) 
expected = expected.unsqueeze(1).type(torch.float) 
Then move them to GPU by .to(device)