[Tensorflow/ Keras] Model Profiler

Tensorflow/ Keras Model Profiler

Gives you some basic but important information about your tf or keras model like,

• Model Parameters
• Model memory requirement on GPU
• Memory required to store parameters model weights.
• GPU availability and GPU IDs if available

 

Sample output

| Model Profile                    | Value         | Unit    |
|----------------------------------|---------------|---------|
| Selected GPUs                    | ['0', '1']    | GPU IDs |
| No. of FLOPs                     | 1.5651        | BFLOPs  |
| GPU Memory Requirement           | 41.7385       | GB      |
| Model Parameters                 | 12.3205       | Million |
| Memory Required by Model Weights | 46.9991       | MB      |

 

Detail :

 

https://pypi.org/project/model-profiler/

model-profiler