Open roboticsai opened 3 years ago
For anyone else who comes across this now after updating fairseq, this can happen because no when using a finetuned model one should set the task as audio_finetuning
For those who are looking for a code to change the task
. In the code below I changed the task from audio_pretraining
to audio_finetuning
.
model_override_rules = {}
model_override_rules['task'] = {'_name': 'audio_finetuning'}
cp_path = os.path.join(BASE_DIR,'pretrained/w2v-large.pt')
model, cfg, task = fairseq.checkpoint_utils.load_model_ensemble_and_task([cp_path], arg_overrides=model_override_rules)
I have trained a wav2vec2 model on custum_dataset of english language. In the model training process i have stoped the training in middle by using ctrn+c after 1 day. After training the model it had created 2 models checkpoint_best.pt checkpoint_last.pt. After this i fine tuned the checkpoint_best.pt with the same database used for training using base_10m.yaml. Here also i have stoped the fine tuning process after 10m. Now when i try to infering the model checkpoint_best.pt i'm geteting this error:
How do i solve this issue? is their anyway to modify the configuration file for training and finetuning such that i do't have to stop the process manually? Is it ok to force stop the process manuallY?