Tensorflow对象检测在开始之前被杀死

时间:2017-06-29 19:05:34

标签: python docker tensorflow

我正在运行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。

1 个答案:

答案 0 :(得分:0)

我现在意识到,一旦我开始,我就会耗尽内存。当我切换到<p style="text-align: center;">Text Here...</p> 并更改ssd_mobilenet_v1_coco_2017_11_17时,一切正常。

但这仍然无法完全解决我的问题。我必须弄清楚如何为通话提供更多内存。