Model 计算模型在keras或tensorflow中使用的精确GPU内存

Model 计算模型在keras或tensorflow中使用的精确GPU内存,model,gpu,Model,Gpu,我想计算keras中的模型内存,以便根据gpu内存可用性调整批处理大小。应如何进行模型内存计算?需要考虑哪些参数?任何简单的公式对我都更有帮助

我想计算keras中的模型内存,以便根据gpu内存可用性调整批处理大小。应如何进行模型内存计算?需要考虑哪些参数?任何简单的公式对我都更有帮助