Error while loading the checkpoint with fairseq
#15
by
Ansu - opened
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!
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?