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

custom keyword - transfer learning #69

Open
Aishaj opened this issue Aug 19, 2021 · 1 comment
Open

custom keyword - transfer learning #69

Aishaj opened this issue Aug 19, 2021 · 1 comment

Comments

@Aishaj
Copy link

Aishaj commented Aug 19, 2021

Hello,

I want to use this trained model as a base model; then, I will retrain the model with a custom dataset. The word that I want to train the model on is not included in the original commands dataset.
Will this transfer learning process work ?
The number of utterances for my keyword is not big.

Thank you

@Burton2000
Copy link
Contributor

Yes this should be possible, you might need to modify the training code a little bit to load the pre-trained checkpoint. Also when loading weights you should try passing in skip_mismatch=True to avoid TF throwing an error when loading the model weights as you will have changed the model structure if you are training for a different number of keywords.

If doing transfer learning I don't think you would need too many samples in your dataset to get decent results.

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

2 participants