Skip to content
This repository has been archived by the owner on Nov 22, 2022. It is now read-only.

Add support for exporting all PyText models that use contextual embeddings #468

Closed
wants to merge 1 commit into from
Closed

Conversation

hikushalhere
Copy link
Contributor

Summary:

  1. Use ONNX Supported operators to inferring shape and reshaping tensors in PretrainedModelEmbedding.
  2. Input name for contextual embeddings are different for RNNG and JointBLSTM models. Fix this.

Differential Revision: D14863549

@facebook-github-bot facebook-github-bot added the CLA Signed Do not delete this pull request or issue due to inactivity. label Apr 12, 2019
…dings (#468)

Summary:
Pull Request resolved: #468

1. Use ONNX Supported operators to inferring shape and reshaping tensors in PretrainedModelEmbedding.
2. Input name for contextual embeddings are different for RNNG and JointBLSTM models. Fix this.

Reviewed By: seayoung1112

Differential Revision: D14863549

fbshipit-source-id: d712ea23a7895b6d4c427cfb07065821d9602d7b
@facebook-github-bot
Copy link
Contributor

This pull request has been merged in cb1782b.

Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Labels
CLA Signed Do not delete this pull request or issue due to inactivity. Merged
Projects
None yet
Development

Successfully merging this pull request may close these issues.

None yet

2 participants