From 34c1ddfecb200e6f77a17cc92a423b7e563e65b8 Mon Sep 17 00:00:00 2001 From: WenmuZhou Date: Tue, 15 Dec 2020 17:54:16 +0800 Subject: [PATCH] Add paddle version notes --- doc/doc_ch/angle_class.md | 2 +- doc/doc_en/angle_class_en.md | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/doc/doc_ch/angle_class.md b/doc/doc_ch/angle_class.md index 857ca514..3f2027b9 100644 --- a/doc/doc_ch/angle_class.md +++ b/doc/doc_ch/angle_class.md @@ -62,7 +62,7 @@ PaddleOCR提供了训练脚本、评估脚本和预测脚本。 *如果您安装的是cpu版本，请将配置文件中的 `use_gpu` 字段修改为false* ``` -# GPU训练 支持单卡，多卡训练，通过 gpus 指定卡号 +# GPU训练 支持单卡，多卡训练，通过 '--gpus' 指定卡号，如果使用的paddle版本小于2.0rc1，请使用'--select_gpus'参数选择要使用的GPU # 启动训练，下面的命令已经写入train.sh文件中，只需修改文件里的配置文件路径即可 python3 -m paddle.distributed.launch --gpus '0,1,2,3,4,5,6,7' tools/train.py -c configs/cls/cls_mv3.yml ``` diff --git a/doc/doc_en/angle_class_en.md b/doc/doc_en/angle_class_en.md index 8bc261af..fcc708be 100644 --- a/doc/doc_en/angle_class_en.md +++ b/doc/doc_en/angle_class_en.md @@ -65,7 +65,7 @@ Start training: ``` # Set PYTHONPATH path export PYTHONPATH=$PYTHONPATH:. -# GPU training Support single card and multi-card training, specify the card number through gpus +# GPU training Support single card and multi-card training, specify the card number through --gpus. If your paddle version has less than 2.0rc1, please use '--selected_gpus' # Start training, the following command has been written into the train.sh file, just modify the configuration file path in the file python3 -m paddle.distributed.launch --gpus '0,1,2,3,4,5,6,7' tools/train.py -c configs/cls/cls_mv3.yml ``` -- GitLab