A C++/CUDA toolkit for neural machine translation (RNN-Based NMT) across multiple GPUs
- CUDA >= 7.5
- CuDNN >= 4
- Gcc >= 4.8
- Boost = 1.58
- Compiling,
cd ./scriptand run./compile_train.shand./compile_test.sh - Training,
cd ./workand run./train_nmt.sh. - Infenrence,
cd ./workand run./test_nmt.sh.
Project leaders: Jiajun Zhang
Project members: Long Zhou, Yang Zhao, and Yining Wang
If you have questions, suggestions and bug reports, please email [email protected] or [email protected].