Skip to content

linzihan-backforward/PyTorchTransformer

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

4 Commits
 
 
 
 
 
 
 
 

Repository files navigation

PyTorchTransformer


Code for a NMT task based on Transformer model and Mutil30kdataset in pytorch


Only contain the train and validate step but not test step


Dataset:Mutil30k de-en Translation Dataset
Proformance:Training accuracy:≈ 89.2%
                Validation accuary:≈62.0%
Hardware: 2*Nvidia Titan Xp
Software: Pytorch 1.1 CUDA 10.0 CuDnn 7.5
Hyperparameters
Training with 150 epoch and all the same hyperparameters as the original model in the paper

Reference

Attention is all you need

About

A simple Transformer model in Pytorch

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages