Skip to content

A Pytorch Implementation of "Attention is All You Need" and "Weighted Transformer Network for Machine Translation"

Notifications You must be signed in to change notification settings

xunge/transformer

 
 

Repository files navigation

A Pytorch Implementation of the Transformer Network

This repository includes pytorch implementations of "Attention is All You Need" (Vaswani et al., NIPS 2017) and "Weighted Transformer Network for Machine Translation" (Ahmed et al., arXiv 2017)

Reference

Paper

  • Vaswani et al., "Attention is All You Need", NIPS 2017
  • Ahmed et al., "Weighted Transformer Network for Machine Translation", Arxiv 2017

Code

Edit in 20210114

fix some bugs to support pytorch 1.7

python train.py -data_path ./preprocess-train.t7 -model_path ./model/ckpt.pth

About

A Pytorch Implementation of "Attention is All You Need" and "Weighted Transformer Network for Machine Translation"

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 100.0%