What’s wrong with training the model on MPS? I am training my WaveNet model using PyTorch on a MacBook with an M1 Pro.
