-
Notifications
You must be signed in to change notification settings - Fork 68
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
Using bert-base-chinese causes CUDA error: device-side assert triggered #129
Comments
it may be vocal size which not matches embedding shape.
|
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
I am following the
Text Classification of MultiNLI Sentences using BERT notebook to interpret BERT classification work. When I changed the line
LANGUAGE = Language.ENGLISH
toLANGUAGE = Language.CHINESE
, indicating to use thebert-base-chinese
pre-trained model, the following error occured:I tried with the original dataset as used in the example notebook, and also tried with my own dataset with Chinese texts. Both resulted in the above error. There was no issue when using multilingual model.
The text was updated successfully, but these errors were encountered: