Open karimkalimu opened 1 year ago
I added an if statement for MPS device because double is not supported for MPS device if mps: .float() else: .double() https://discuss.pytorch.org/t/typeerror-cannot-convert-a-mps-tensor-to-float64-dtype-as-the-mps-framework-doesnt-support-float64-please-use-float32-instead/180852
I added an if statement for MPS device because double is not supported for MPS device if mps: .float() else: .double() https://discuss.pytorch.org/t/typeerror-cannot-convert-a-mps-tensor-to-float64-dtype-as-the-mps-framework-doesnt-support-float64-please-use-float32-instead/180852