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

时间:2019-10-07 03:03:08

标签: tensorflow keras

我已经在Keras中开发了第一层,当我运行基于它的模型时,它花费了很长时间,然后提示:

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

此后,根据我设置的批处理大小,它可能会引发OOM(内存不足)错误,也可能不会引发。如果没有,则训练开始。但是由于要花大约30分钟的时间来,因此对于我来说要调整最佳的批量大小并不容易。

有人知道什么使内存优化器花费这么长时间吗?我该如何解决这个问题?

如果您想查看图层的代码,可以here找到它。

0 个答案:

没有答案