PyTorch Forums
Cannot convert a MPS Tensor to float64 dtype as the MPS framework doesn't support float64. Please use float32 instead
soulitzer
October 1, 2023, 3:58am
2
You might be looking for
DeviceCastTransform — torchrl main documentation
?
show post in topic