johnnyworker1012 / neural-machine-translation

Assignment 4 (NMT)

Note: Heavily inspired by the https://github.com/pcyin/pytorch_nmt repository

Create virtual environment

conda env create --file local_env.yml

Activate and deactivate environment

conda activate local_nmt
conda deactivate

Install necessary packages (On your VM)

pip install -r gpu_requirements.txt

Generate the required vocab files (Exce

sh run.sh vocab

Train model on your local machine

sh run.sh train_local