model - Calculate the exact GPU memory taken by the model in keras or tensorflow

Question

I want to calculate the model memory in keras, so that I can tune the batch size based on the gpu memory availability. How the model memory calculation should be done? What are the parameters need to be taken into consideration? Any simple formula is really more helpful for me.

标签： modelgpu