Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

fine-tuning pre train model #28

Open
Moonmore opened this issue May 31, 2024 · 3 comments
Open

fine-tuning pre train model #28

Moonmore opened this issue May 31, 2024 · 3 comments

Comments

@Moonmore
Copy link

Hi, thank you very much for your work.

I want to continue to do some interesting work based on your work.
I have not found any related model fine-tuning on modelscore and github.
Can you please guide me on how to use your model for model fine-tuning and retraining?

many thanks

@ddlBoJack
Copy link
Owner

Hi, you can fine-tune the model with the FunASR auto model pipeline.

@buanide
Copy link

buanide commented Jul 17, 2024

this is what you call fine-tuning ?


model = AutoModel(model="iic/emotion2vec_base_finetuned") # Alternative: iic/emotion2vec_plus_seed, iic/emotion2vec_plus_base, iic/emotion2vec_plus_large and iic/emotion2vec_base_finetuned

wav_file = f"{model.model_path}/example/test.wav"
rec_result = model.generate(wav_file, output_dir="./outputs", granularity="utterance", extract_embedding=False)


?

@Zevrap-81
Copy link

I want to load the model and finetune it myself for better accuracy on my voice and the voice of my colleagues.

I see that you provided the code for the upstream model and the model checkpoints i get from the emotion2vec_base.pt.

However, the checkpoint does not fully match the model. Specifically i get a RuntimeError

RuntimeError: Error(s) in loading state_dict for Data2VecMultiModel:
        Unexpected key(s) in state_dict: "modality_encoders.AUDIO.extra_tokens", "modality_encoders.AUDIO.alibi_scale". 

I think there are many such missmatches later as well once i resolve this. So to save myself some time, could you provide me the corresponding configs?

thanks in advance

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

4 participants