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 ./script
and run./compile_train.sh
and./compile_test.sh
- Training,
cd ./work
and run./train_nmt.sh
. - Infenrence,
cd ./work
and 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 wszlong@gmail.com or long.zhou@nlpr.ia.ac.cn.