重新训练对象检测模型-内存不足（OOM）杀手-Google Coral

Question

我正尝试按照下面的链接为Google Coral Accelerator训练对象检测模型

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

主机系统基于具有docker环境的Linux Mint

• CPU：2.00GHz的Intel（R）Core（TM）i3-5005U CPU
• 图形：卡：Intel HD Graphics 5500
• 操作系统：Linux Mint 19 Tara
• 内存大小：8G

但是在开始培训工作之后

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

该过程被OOM杀手杀死

./ retrain_detection_model.sh：第45行：86被杀死
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}”

感谢您的帮助！

Answer 1

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