Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
wangsen
paddle_dbnet
Commits
7de103c5
Commit
7de103c5
authored
Dec 15, 2020
by
LDOUBLEV
Browse files
update doc
parent
b3807c2f
Changes
3
Hide whitespace changes
Inline
Side-by-side
Showing
3 changed files
with
11 additions
and
5 deletions
+11
-5
doc/doc_ch/detection.md
doc/doc_ch/detection.md
+2
-2
doc/doc_en/detection_en.md
doc/doc_en/detection_en.md
+4
-2
train.sh
train.sh
+5
-1
No files found.
doc/doc_ch/detection.md
View file @
7de103c5
...
...
@@ -76,8 +76,8 @@ tar -xf ./pretrain_models/MobileNetV3_large_x0_5_pretrained.tar ./pretrain_model
# 单机单卡训练 mv3_db 模型
python3 tools/train.py
-c
configs/det/det_mv3_db.yml
\
-o
Global.pretrain_weights
=
./pretrain_models/MobileNetV3_large_x0_5_pretrained/
# 单机多卡训练,通过 --
select_
gpus 参数设置使用的GPU ID;
python3
-m
paddle.distributed.launch
--
selected_
gpus
'0,1,2,3'
tools/train.py
-c
configs/det/det_mv3_db.yml
\
# 单机多卡训练,通过 --gpus 参数设置使用的GPU ID;
如果使用的paddle版本小于2.0rc1,请使用'--select_gpus'参数选择要使用的GPU
python3
-m
paddle.distributed.launch
--gpus
'0,1,2,3'
tools/train.py
-c
configs/det/det_mv3_db.yml
\
-o
Global.pretrain_weights
=
./pretrain_models/MobileNetV3_large_x0_5_pretrained/
```
...
...
doc/doc_en/detection_en.md
View file @
7de103c5
...
...
@@ -76,8 +76,10 @@ You can also use `-o` to change the training parameters without modifying the ym
python3 tools/train.py
-c
configs/det/det_mv3_db.yml
-o
Optimizer.base_lr
=
0.0001
# multi-GPU training
# Set the GPU ID used by the '--select_gpus' parameter;
python3
-m
paddle.distributed.launch
--selected_gpus
'0,1,2,3'
tools/train.py
-c
configs/det/det_mv3_db.yml
-o
Optimizer.base_lr
=
0.0001
# Set the GPU ID used by the '--gpus' parameter; If your paddle version less than 2.0rc1, please use '--selected_gpus'
python3
-m
paddle.distributed.launch
--gpus
'0,1,2,3'
tools/train.py
-c
configs/det/det_mv3_db.yml
-o
Optimizer.base_lr
=
0.0001
```
#### load trained model and continue training
...
...
train.sh
View file @
7de103c5
python3
-m
paddle.distributed.launch
--selected_gpus
'0,1,2,3,4,5,6,7'
tools/train.py
-c
configs/rec/rec_mv3_none_bilstm_ctc.yml
\ No newline at end of file
# for paddle.__version__ >= 2.0rc1
python3
-m
paddle.distributed.launch
--gpus
'0,1,2,3,4,5,6,7'
tools/train.py
-c
configs/rec/rec_mv3_none_bilstm_ctc.yml
# for paddle.__version__ < 2.0rc1
# python3 -m paddle.distributed.launch --selected_gpus '0,1,2,3,4,5,6,7' tools/train.py -c configs/rec/rec_mv3_none_bilstm_ctc.yml
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment