我可以使用Keras测量网络在训练/推理期间需要多少内存？

Question

我想知道一个给定的培训设置（模型，优化器，批量大小）我的网络需要多少内存。 Can Keras会自动给我这个号码吗？

同样有趣的是：在推理过程中需要多少内存？