Skip to content

Commit ebee656

Browse files
authored
[cherry-pick] update docs with cpu training (#2926)
1 parent ab46567 commit ebee656

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

PaddleCV/PaddleDetection/docs/GETTING_STARTED.md

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -20,13 +20,16 @@ python tools/train.py -c configs/faster_rcnn_r50_1x.yml
2020

2121
```bash
2222
export CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7
23+
# or run on CPU with:
24+
# export CPU_NUM=8
2325
python tools/train.py -c configs/faster_rcnn_r50_1x.yml
2426
```
2527

2628
- Datasets is stored in `dataset/coco` by default (configurable).
2729
- Pretrained model is downloaded automatically and cached in `~/.cache/paddle/weights`.
2830
- Model checkpoints is saved in `output` by default (configurable).
2931
- To check out hyper parameters used, please refer to the config file.
32+
- RCNN models training on CPU is not supported on PaddlePaddle<=1.5.1 and will be fixed on later version.
3033

3134
Alternating between training epoch and evaluation run is possible, simply pass
3235
in `--eval=True` to do so (tested with `SSD` detector on Pascal-VOC, not

0 commit comments

Comments
 (0)