This repository has been archived by the owner on Nov 22, 2022. It is now read-only.
Add torchscriptify API in tokenizer and tensorizer #1055
Closed
Conversation
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Summary: implement BertTensoriimplement BertTensorizer and RoBERTaTensorizer in TorchScript. ScriptTensorizer have two APIs 1. numberize: process a single line of input (single string for classification and a pair of string for pair classification), the output will be a list of token ids (e.g token index in the vocab) 2. tensorize: process multiple of line of input, calling numberize and batch all the result together, generate the output tensor as the model input Differential Revision: D17941983 fbshipit-source-id: 56904c232ff3478c97a31736be20f5d464489e59
Summary: Add torchscriptify API in tokenizer and tensorizer Differential Revision: D17943029 fbshipit-source-id: 1d5883028a0e23f4c7f48803ad59fb2128dd941a
facebook-github-bot
added
the
CLA Signed
|
This pull request was exported from Phabricator. Differential Revision: D17943029 |
|
This pull request has been merged in 6d2bc1e. |
Sign up for free
to subscribe to this conversation on GitHub.
Already have an account?
Sign in.
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
Summary: Add torchscriptify API in tokenizer and tensorizer
Differential Revision: D17943029