RuntimeError: Found dtype Double but expected Float Exception raised from compute_types at …aten/src/ATen/native/TensorIterator.cpp:183

Question

For the following code block:

##torch.autograd.set_detect_anomaly(True)
network = Network()
network.cuda()    

criterion = nn.MSELoss()
optimizer = optim.Ada