Skip to content

juliusshufan/pytorch_rnn_benchmarks

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

11 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

pytorch-rnn-benchmark


A program to test performance of several RNNs(realized by Pytorch) on different hardware platforms, providing benchmark for further optimization.

Requirements


  • Pytorch

Usage


bash run.sh platform-name [train] [daily]

platform-name can be 'bdw', 'skx', 'knm', 'knl', 'gpu'.
If you want to test performance of training, please add parameter train.
If you want to run daily test(just test 2 size), please add parameter daily.

Benchmark size info

For LSTM/RNN/GRU, we will follow the size of Baidu DeepBench. And for LSTM we wiil add OpenNMT size as an supplement.

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published