一尘不染

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

docker

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

我相信我所有的代码和代码都在正确的地方。但是,当我尝试重新训练时,tensorflow在开始训练之前会自行杀死,但不会出现任何问题或错误。

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

我想我有东西出没了,但是没有任何错误或输出,我不知道该看哪里!

我正在按照此处的指南在本地运行事情:link。pet数据是从同一GitHub:link获得的。我也从同一GitHub获得了模型配置。链接

我选择了inception_v2。


阅读 325

收藏
2020-06-17

共1个答案

一尘不染

现在,我意识到我一开始就只是内存不足。当我切换ssd_mobilenet_v1_coco_2017_11_17并更改时batch_size = 1,一切正常。

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

2020-06-17