NMT models for Bergamot
Teacher and student NMT models are described in subdirectories. We have prepared models for the following language pairs:
Download and compile the latest marian-dev in the upper directory:
The compilation requires MKL and gperftools, see the Marian documentation to find which packages are needed on Ubuntu.
Go to a subdirectory, download data and model files, e.g.:
cd eten bash download-data.sh bash download-models.sh
See README file for more details.
Go to a model directory and run the evaluation script, e.g.:
cd eten.student.tiny11 bash eval.sh -d 0 # -d 0 means use first GPU
or benchmarking scripts:
bash speed.gpu.sh -d 0 bash speed.cpu.sh bash speed.cpu.packed8.sh # requires CPU with avx2
See individual scripts for details.