我正在运行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。
现在,我意识到我一开始就只是内存不足。当我切换ssd_mobilenet_v1_coco_2017_11_17并更改时batch_size = 1,一切正常。
ssd_mobilenet_v1_coco_2017_11_17
batch_size = 1
这仍然不能完全解决我的问题。我将不得不弄清楚如何为通话提供更多的内存。