-
-
Notifications
You must be signed in to change notification settings - Fork 462
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
Cannot use community contributed BERT model? #196
Comments
I tried this way:
but got:
|
I just saw your upgrades ''Upgraded to use AutoModel and AutoTokeizer for ContextualWordEmbsAug, ContextualWordEmbsForSentenceAug and AbstSummAug." in Sep 2020 |
NLPAug uses model name to check model type (different model type has different behavior). Therefore, "Model name value is unexpected. Only support BERT, DistilBERT, RoBERTa and XLNet model." is thrown. For the second expcetion, it is caused by model type as well. |
Can try dev version (1.1.2dev) first by By provide "model_type" (it is "bert" in your case) parameter, it should fix the problem
|
It works, thanks a lot! |
Hi!
I want to use a community-contributed pre-trained BERT model PubMedBERT on biomedical literature.
I ran the following code:
and I got:
Is it because 'ContextualWordEmbsAug' does not support 'PubMedBERT' or I used wrong code?
If it does not support it, could you make it possible?
Thank you.
The text was updated successfully, but these errors were encountered: