为什么 Tensorfow 需要很长时间才能使用内存优化器？

Question

我已经在 Keras 中开发了我的第一层，当我运行基于它的模型时，它需要很长时间然后提示：

tensorflow/core/grappler/optimizers/meta_optimizer.cc:499] memory_optimizer failed: Deadline exceeded: memory_optimizer exceeded deadline., time = 290379.781ms.

之后，根据我设置的批量大小，它可能会抛出 OOM（内存不足）错误。如果没有，则训练开始。但是由于出现错误需要大约 30 分钟，因此我很难调整可能的最佳批量大小。

有谁知道是什么让内存优化器需要这么长时间？我该如何解决这个问题？

如果您有兴趣查看图层的代码，可以在此处找到。