从python中访问tensorflow/core/framework/cpu\u分配器\u impl.cc

Question

我发现了很多关于tensorflow警告的问题

tensorflow/core/framework/cpu_allocator_impl.cc:81] Allocation of
xxxxxxxxx exceeds 10% of system memory

我知道这只是一个警告，而不是一个错误，它的显示可以被抑制。我也知道如何（很可能）解决这个问题：减少批量大小

然而，我从未发现问题的解决方式与此相反：

考虑到我的网络和它运行的系统的当前状态，它可以容纳的最大批量是多少

那么，有没有一种方法可以从python（3.7）接口访问tensorflow在内部所做的事情呢

我正在CPU上运行TF2.2.0。我知道有一些方法可以限制GPU内存的使用（让它增长/不使用100%的可用空间/etc），但我还没有找到CPU的等效方法。CPU（https://www.tensorflow.org/api_docs/python/tf/config/set_logical_device_configuration）不支持创建具有内存限制的逻辑设备