Skip to content

MFajcik/Transformer_playground

Repository files navigation

Transformer playground

This is my playground with Transformer seq2seq model implementation in PyTorch. Implementation is basically taken from excellent The Annotated Transformer guide, refactored, updated for pytorch 0.4 and more commented.

Update:

  • BLEU evaluation implementation has been added.
  • Beam Search implementation inspired by implementation in openNMT have been added.

About

Transformer network playground

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages