CODEWITHSUNDEEP
pytorchspeech-recognitionhuggingface-transformerstorchapple-m2
user1680859
user1680859

Reputation: 1194

WER different for Colab and Apple M2 Max

I am going through this blog: https://huggingface.co/blog/fine-tune-xlsr-wav2vec2 aimed for speech recognition. When I run the code in Google Colab, it works fine, WER is lower then 0.5. But when I try the same code with MacBook Pro (Apple M2 max), I get WER equal to 1.0. it seems like it has to do something of how numbers are treated, but cannot understand how to have the same results. Predictions are empty strings. The only change that I make is on the model:

mps_device = torch.device("mps")
model.to(mps_device)

Upvotes: 0

Views: 153

Answers (1)

user1680859
user1680859

Reputation: 1194

I moved everything to cpu (torch, model, training arguments) and it worked fine. Much slower but correct WER. It seems Apple's gpu/mps can not process the correct training.

Upvotes: 0

Related Questions