/transformer-translator

A C++/CUDA toolkit for Transformer (NMT) Translator (Decoder)

Primary LanguageC++

Transformer-translator

This project is a simple implementation of C++/CUDA code for Transformer(NMT) Decoder.

Requirements

  1. CUDA >= 7.5
  2. CuDNN >= 4
  3. Gcc >= 4.8
  4. Boost = 1.58

Usage

  1. Employ Transformer to train a NMT model.
  2. Employ Transformer-translator to decode (run ./test.sh).

Contact

If you have questions, suggestions and bug reports, please email wszlong@gmail.com or long.zhou@nlpr.ia.ac.cn.