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

SPARKNLP 801 set up warmup for all embeddings #13851

Conversation

maziyarpanahi
Copy link
Member

@maziyarpanahi maziyarpanahi commented Jun 8, 2023

This PR adds warmup TF session for the remaining of annotators:

  • ALBERT
  • BERT
  • CamemBERT
  • DistilBERT
  • RoBERTa
  • XLM-RoBERTa
  • XLNet

The warmup runs once when the model is being loaded to reduce the first inference time, it is also useful when external models are being imported. This makes sure those models work right there without the need of inferencing afterwards.

@maziyarpanahi maziyarpanahi self-assigned this Jun 8, 2023
@maziyarpanahi maziyarpanahi changed the base branch from master to release/444-release-candidate June 8, 2023 14:44
@maziyarpanahi maziyarpanahi added new-feature Introducing a new feature DON'T MERGE Do not merge this PR labels Jun 8, 2023
@maziyarpanahi maziyarpanahi merged commit adf5586 into release/444-release-candidate Jun 8, 2023
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
DON'T MERGE Do not merge this PR new-feature Introducing a new feature
Projects
None yet
Development

Successfully merging this pull request may close these issues.

1 participant