我已经在Keras中开发了第一层,当我运行基于它的模型时,它花费了很长时间,然后提示:
tensorflow/core/grappler/optimizers/meta_optimizer.cc:499] memory_optimizer failed: Deadline exceeded: memory_optimizer exceeded deadline., time = 290379.781ms.
此后,根据我设置的批处理大小,它可能会引发OOM(内存不足)错误,也可能不会引发。如果没有,则训练开始。但是由于要花大约30分钟的时间来,因此对于我来说要调整最佳的批量大小并不容易。
有人知道什么使内存优化器花费这么长时间吗?我该如何解决这个问题?
如果您想查看图层的代码,可以here找到它。