Error while loading the checkpoint with fairseq

#15

by Ansu - opened Feb 20

Discussion

Ansu

Feb 20

•

edited Feb 20

Hi!

I am trying to make a continuous-pretrain from this checkpoint, but I am struggling to load the .pt checkpoint with your fork of Fairseq. I get this error:

AssertionError: Could not infer task type from {'_name': 'mhubert_pretraining' ...}

It seems that there is not a task named 'mhubert_pretrain'. Do you know how can I solve this?

Thank you in advance!

mzboito

UTTER - Unified Transcription and Translation for Extended Reality org Feb 25

Hi Ansu,

Thanks for the interest in the model. This is curious, because the task should indeed be there: https://github.com/utter-project/fairseq/blob/2959684ecc6e28e34a075e2f1a08bba100e11743/fairseq/tasks/hubert_pretraining.py#L293

Did you make sure to pip install -e . inside the fairseq fork repository before trying to run the training?

Ansu

Apr 5

Hi,
Sorry for the late response. Yeah, that was the problem, I had installed the original fairseq as I couldn't install your fork, but I got it working.
Thanks!

mzboito changed discussion status to closed Apr 5

Upload images, audio, and videos by dragging in the text input, pasting, or clicking here.

Tap or paste here to upload images

· Sign up or log in to comment