python - Tensorflow 对象检测在启动前被杀死

Question

我正在运行 docker image tensorflow:1.1.0。我通过在本地克隆它并为我的 docker 提供到该文件夹​​的连接来添加 tensorflow 对象检测 api github。我正在尝试重新创建他们的宠物示例。

我相信我拥有所有代码并且代码都在正确的位置。但是，当我尝试重新训练时，tensorflow 在开始训练之前自行终止，但没有给出任何问题或错误。

INFO:tensorflow:Starting Session.
INFO:tensorflow:Starting Queues.
INFO:tensorflow:global_step/sec: 0
Killed

我想我有一些错误或命令，但没有任何错误或输出我不知道去哪里找!

我正在按照此处的指南在本地运行:link .宠物数据是从同一个 GitHub 获得的:link .我也从同一个 GitHub 获得了我的模型配置。 link

我选择 inception_v2。

Answer 1

我现在意识到，我一开始就用完了内存。当我切换到 ssd_mobilenet_v1_coco_2017_11_17 并更改 batch_size = 1 时，一切正常。

但这仍然不能完全解决我的问题。我将不得不弄清楚如何为调用提供更多内存。