/pytorch-seq2seq

An LSTM-based implementation of sequence to sequence learning, in PyTorch.

Primary LanguagePythonMIT LicenseMIT

pytorch seq2seq

This repository contains an implementation of an LSTM sequence to sequence model in PyTorch.

examples:

  • German to English machine translation
  • Video captioning on the Yahoo TGIF dataset

requirements

Latest versions of the following:

  • Python 3.5
  • PyTorch
  • spaCy
  • unidecode
  • numpy

contributing

Let me know if you spot any issues, and please submit a pull request if you have suggestions!!