python-repository-hub / seq2seq-pytorch-example

A toy example of sequence to sequence (seq2seq) model with pytorch.

A toy example of sequence to sequence (seq2seq) model with PyTorch.

Toy Dataset

• Source Sequence: Consecutive numbers in 0-9 with length 10
• Target Sequence: Consecutive numbers which add 1 to each source sequence number
• Example:

Source: 3 4 5 6 7 8 9 0 1 2
Target: 4 5 6 7 8 9 0 1 2 3

Train & Run

python translate.py

Trained model will be persisted in model.dat. Remove the file to re-train.