Hi,
great, that fixed it!
I didn’t do it because I was using instead
torch.set_default_dtype(torch.float64)

and I understood that the effect of that was to make torch always use double precision tensors. If not, what is that command doing?
Thanks a lot!
Helios