-
Notifications
You must be signed in to change notification settings - Fork 83
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
Does cuBERT support sequence tagging output? #7
Comments
Currently, we only support standard BERT graph and standard BERT outputs, in your case the most related output type I guess is You can fork our repository and make some modification if you want [batch_size * sequence_length] (I guess you put some additional layers after standard BERT). Or you can also get our You can have a look at |
I am working on a sequence tagging task, where the logits output should be [batchsize * sequence length]. Does cuBERT support that?
The text was updated successfully, but these errors were encountered: