Skip to content

Latest commit

 

History

19 Commits

Folders and files

NameName
Last commit message
Last commit date
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Comparsion of TensorFlow Wrappers

Run Keras, TensorLayer and Tflearn with same model and data on a same GPU machine.

The parameter initialization may have slightly different, but would not effect the speed.

Feel free to PUSH !

Speed of MLP

GPU: GTX980

TensorFlow: r0.10

Data: MNIST train:50k val:10k test:10k

Model: 784-800-800-10

Num of epochs: 200

Batch size: 500

Keras: 282.475250s = 1.41 s/epoch

TensorLayer: 116.670947s = 0.58 s/epoch

Tflearn:

Speed of CNN

Speed of LSTM

About

Comparison of TensorFlow Wrappers

Topics

Resources

Stars

9 stars

Watchers

0 watching

Forks

Releases

Packages

Contributors

Languages