重新训练对象检测模型 - 内存不足 (OOM) 杀手 - Google 珊瑚

Retrain an object detection model - Out of memory (OOM) Killer - Google coral

我正在尝试 根据以下 Google 珊瑚加速器 重新训练对象检测模型 link

https://coral.ai/docs/edgetpu/retrain-detection/#prerequisites

主机系统基于 Linux Mint 和 docker 环境

CPU：英特尔(R) 酷睿(TM) i3-5005U CPU @ 2.00GHz
显卡：显卡：Intel HD Graphics 5500
OS : Linux 薄荷 19 塔拉
内存大小：8G

但在开始训练作业后

root@beaa5d65a1d5:/tensorflow/models/research# ./retrain_detection_model.sh --num_training_steps ${NUM_TRAINING_STEPS} --num_eval_steps ${NUM_EVAL_STEPS}

进程被OOM killer杀死

./retrain_detection_model.sh: line 45: 86 Killed
python object_detection/model_main.py --pipeline_config_path="${CKPT_DIR}/pipeline.config" --model_dir="${TRAIN_DIR}" --num_train_steps="${num_training_steps}" --num_eval_steps="${num_eval_steps}"

感谢任何帮助！

这是由于硬件限制导致的内存不足问题。您可以做的两件事是添加更多 RAM 或交换空间（使用存储作为 RAM）。虽然跟后者会很慢

重新训练对象检测模型 - 内存不足 (OOM) 杀手 - Google 珊瑚

Retrain an object detection model - Out of memory (OOM) Killer - Google coral

tensorflow

google-coral