Neural Machine Translation
Encoder-Decoder architecture for neural machine translation BN->EN
.
language.py
and trainer trainer.py
class.Install the following if not installed.
data
folder check out the sample.txt
for data format.experiment.ini
i.e. lang1, rnn, hidden_size
.reverse
a bool
variable will change the model training i.e. BN->EN
to EN->BN
python train.py
For resume training change the obj_path
from false
to logs
and checkpoint
false
to model path i.e. path/to/model.pt
.
For train in new language you only need to change the tokenizer
in train.py
. You can use spacy
tokenizer or custom made tokernizer.