TensorFlow對象檢測限制內存和CPU使用

Question

我設法運行教程中的tensorflow寵物示例。我決定使用最慢的模型（因爲我想用我自己的數據）。但是，當我開始訓練時，跑步後會死亡。它使用我所有的cpus（4）和我所有的內存8GB。你是否知道我可以將CPU數量限制爲2並限制使用的內存量？如果我減少批量大小？我的批量大小已經是1

我設法通過減少大小調整運行：

image_resizer { keep_aspect_ratio_resizer { MIN_DIMENSION：300 MAX_DIMENSION：612 }

在此先感謝。

Answer 1

減少內存使用量的另一個想法是減少輸入數據的隊列大小。具體而言，在object_detection/protos/train.proto文件中，您將看到batch_queue_capacity和prefetch_queue_capacity的條目---考慮在配置文件中將這些字段明確設置爲較小的數字。