TensorFlow对象检测限制内存和CPU使用

Question

我设法从教程中运行tensorflow pet示例。我决定使用最慢的模型（因为我想用于我自己的数据）。然而，当我开始训练时，它会在运行一段时间后被杀死。它使用了我所有的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的条目---请考虑在配置文件中将这些字段明确设置为较小的数字。