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

Update nmt_with_attention #626

Merged
merged 1 commit into from
May 25, 2021
Merged

Update nmt_with_attention #626

merged 1 commit into from
May 25, 2021

Conversation

tf-text-github-robot
Copy link
Collaborator

Update nmt_with_attention

For TF2.5

  • Use the TextVectorization layer.
  • Use the AdditiveAttention layer.
  • tf.function the translate loop for text->text export.
  • Add more inline explanations, and sanity checks.
  • Add shape assertions throughout the code to make it easier to follow.

Fixes: tensorflow/tensorflow#38248
Fixes: tensorflow/tensorflow#39654
See also: tensorflow/tensorflow#49237

@review-notebook-app
Copy link

Check out this pull request on  ReviewNB

See visual diffs & provide feedback on Jupyter Notebooks.


Powered by ReviewNB

@google-cla google-cla bot added the cla: yes label May 25, 2021
@tf-text-github-robot tf-text-github-robot force-pushed the test_370250185 branch 2 times, most recently from 0d2a2fa to 1a98c05 Compare May 25, 2021 00:37
For TF2.5

- Use the TextVectorization layer.
- Use the AdditiveAttention layer.
- tf.function the translate loop for text->text export.
- Add more inline explanations, and sanity checks.
- Add shape assertions throughout the code to make it easier to follow.

Fixes: tensorflow/tensorflow#38248
Fixes: tensorflow/tensorflow#39654
See also: tensorflow/tensorflow#49237
PiperOrigin-RevId: 375597559
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
Projects
None yet
2 participants