From ebee6565e4bfb379779f6df4a0bb4c1e5d11b88a Mon Sep 17 00:00:00 2001 From: wangguanzhong Date: Thu, 25 Jul 2019 16:45:52 +0800 Subject: [PATCH] [cherry-pick] update docs with cpu training (#2926) --- PaddleCV/PaddleDetection/docs/GETTING_STARTED.md | 3 +++ 1 file changed, 3 insertions(+) diff --git a/PaddleCV/PaddleDetection/docs/GETTING_STARTED.md b/PaddleCV/PaddleDetection/docs/GETTING_STARTED.md index b532aa22..544e0a1c 100644 --- a/PaddleCV/PaddleDetection/docs/GETTING_STARTED.md +++ b/PaddleCV/PaddleDetection/docs/GETTING_STARTED.md @@ -20,6 +20,8 @@ python tools/train.py -c configs/faster_rcnn_r50_1x.yml ```bash export CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7 +# or run on CPU with: +# export CPU_NUM=8 python tools/train.py -c configs/faster_rcnn_r50_1x.yml ``` @@ -27,6 +29,7 @@ python tools/train.py -c configs/faster_rcnn_r50_1x.yml - Pretrained model is downloaded automatically and cached in `~/.cache/paddle/weights`. - Model checkpoints is saved in `output` by default (configurable). - To check out hyper parameters used, please refer to the config file. +- RCNN models training on CPU is not supported on PaddlePaddle<=1.5.1 and will be fixed on later version. Alternating between training epoch and evaluation run is possible, simply pass in `--eval=True` to do so (tested with `SSD` detector on Pascal-VOC, not -- GitLab