Skip to content

Latest commit

 

History

History
32 lines (16 loc) · 507 Bytes

README.md

File metadata and controls

32 lines (16 loc) · 507 Bytes

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