Skip to content

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

Notifications You must be signed in to change notification settings

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 [email protected] or [email protected].

About

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

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published