/Pytorch-Transformer

PyTorch implementation of Transformer as described in "Attention Is All You Need"

Primary LanguageJupyter Notebook

PyTorch Transformer

PyTorch implementation of the Transformer as described in "Attention Is All You Need"

Running

  • Download and place this dataset (or any other similar one) in ./data
  • Run translation.ipynb
  • Have fun burning your gpu :)