Skip to content

wszlong/transformer-translator

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

8 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

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.

About

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

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published