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

faster prediction using GoEmotions-pytorch models based on bert-mini, bert-small or bert-tiny #4

Open
chedia-d opened this issue Jan 13, 2021 · 1 comment

Comments

@chedia-d
Copy link

Hi,
Thank you for your great work on GoEmotions-pytorch!
I am trying to use your code to generate models using either bert-mini, bert-small or bert-tiny for faster predictions.
I changed the file original.json by setting model_name_or_path to prajjwal1/bert-mini for example and I run python3 run_goemotions.py --taxonomy original
It works and the new model is a bit faster than the one using bert-base. However, I was wondering if I need to also change the tokenizer_name_or_path to a different value. The original value is "monologg/bert-base-cased-goemotions-original". Any thoughts on how to get a tokenizer based on bert-mini?

Many thanks!
Chedia

@monologg
Copy link
Owner

If you use the vocab from prajjwal/bert-mini when training, then you should also change the tokenizer_name_or_path

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