yjk15133895098
/
tensorlayer3
forked from TensorLayer/tensorlayer3
1
0
Fork 0
Code Issues Pull Requests 1 Releases Wiki Activity
tensorlayer3/tests/performance_test/vgg
History
Eric Lai 370fdff9da update 2021-05-11 14:32:09 +08:00
..
README.md update 2021-05-11 14:32:09 +08:00
__init__.py update 2021-05-11 14:32:09 +08:00
exp_config.py update 2021-05-11 14:32:09 +08:00
keras_test.py update 2021-05-11 14:32:09 +08:00
pytorch_test.py update 2021-05-11 14:32:09 +08:00
tf2-autograph.py update 2021-05-11 14:32:09 +08:00
tf2-eager.py update 2021-05-11 14:32:09 +08:00
tl2-autograph.py update 2021-05-11 14:32:09 +08:00
tl2-eager.py update 2021-05-11 14:32:09 +08:00
tl2-static-autograph.py update 2021-05-11 14:32:09 +08:00
tl2-static-eager.py update 2021-05-11 14:32:09 +08:00

README.md

Performance Test (VGG16)

Introduction

This test compares performance of the following libraries:

  1. TensorLayer v. 2.0
  2. TensorFlow (tf.keras) v. 2.0.0-alpha
  3. Keras v. 2.2.4

With GPU

Hardware:

  • CPU: Intel(R) Xeon(R) CPU E5-2630 v4 @ 2.20GHz 40 core
  • GPU: TITAN Xp

Experiment Settings:

  • Model: VGG16
  • Batch size: 32
  • Number of iterations: 300

Results:

Mode Lib Data Format Max GPU Memory Usage(MB) Max CPU Memory Usage(MB) Avg CPU Memory Usage(MB) Runtime (sec)
AutoGraph TensorFlow 2.0 channel last 11833 2161 2136 74
Tensorlayer 2.0 channel last 11833 2187 2169 76
Graph Keras channel last 8677 2580 2576 101
Eager TensorFlow 2.0 channel last 8723 2052 2024 97
TensorLayer 2.0 channel last 8723 2010 2007 95