Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
weixin_41840029
PaddleOCR
提交
19c5ab23
P
PaddleOCR
项目概览
weixin_41840029
/
PaddleOCR
与 Fork 源项目一致
Fork自
PaddlePaddle / PaddleOCR
通知
1
Star
1
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
0
列表
看板
标记
里程碑
合并请求
0
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
P
PaddleOCR
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
0
Issue
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
提交
19c5ab23
编写于
1月 20, 2022
作者:
L
LDOUBLEV
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
delete extra params and set profile_option as bool
上级
d62b3d8f
变更
3
显示空白变更内容
内联
并排
Showing
3 changed file
with
10 addition
and
12 deletion
+10
-12
test_tipc/configs/det_mv3_db_v2.0/train_benchmark.txt
test_tipc/configs/det_mv3_db_v2.0/train_benchmark.txt
+1
-1
test_tipc/test_train_inference_python.sh
test_tipc/test_train_inference_python.sh
+3
-9
tools/program.py
tools/program.py
+6
-2
未找到文件。
test_tipc/configs/det_mv3_db_v2.0/train_benchmark.txt
浏览文件 @
19c5ab23
...
...
@@ -10,7 +10,7 @@ Train.loader.batch_size_per_card:benchmark_train=16
Global.pretrained_model:null
train_model_name:latest
train_infer_img_dir:null
null:null
--profiler_options:True
##
trainer:norm_train
norm_train:tools/train.py -c configs/det/det_mv3_db.yml -o Global.pretrained_model=./pretrain_models/MobileNetV3_large_x0_5_pretrained
...
...
test_tipc/test_train_inference_python.sh
浏览文件 @
19c5ab23
...
...
@@ -5,12 +5,6 @@ FILENAME=$1
# MODE be one of ['lite_train_lite_infer' 'lite_train_whole_infer' 'whole_train_whole_infer', 'whole_infer', 'klquant_whole_infer']
MODE
=
$2
if
[
$#
-eq
3
]
;
then
extra_train_params
=
$3
else
extra_train_params
=
""
fi
dataline
=
$(
awk
'NR==1, NR==51{print}'
$FILENAME
)
# parser params
...
...
@@ -336,11 +330,11 @@ else
set_save_model
=
$(
func_set_params
"
${
save_model_key
}
"
"
${
save_log
}
"
)
if
[
${#
gpu
}
-le
2
]
;
then
# train with cpu or single gpu
cmd
=
"
${
python
}
${
run_train
}
${
set_use_gpu
}
${
set_save_model
}
${
set_epoch
}
${
set_pretrain
}
${
set_autocast
}
${
set_batchsize
}
${
set_train_params1
}
${
set_amp_config
}
${
extra_train_params
}
"
cmd
=
"
${
python
}
${
run_train
}
${
set_use_gpu
}
${
set_save_model
}
${
set_epoch
}
${
set_pretrain
}
${
set_autocast
}
${
set_batchsize
}
${
set_train_params1
}
${
set_amp_config
}
"
elif
[
${#
ips
}
-le
26
]
;
then
# train with multi-gpu
cmd
=
"
${
python
}
-m paddle.distributed.launch --gpus=
${
gpu
}
${
run_train
}
${
set_use_gpu
}
${
set_save_model
}
${
set_epoch
}
${
set_pretrain
}
${
set_autocast
}
${
set_batchsize
}
${
set_train_params1
}
${
set_amp_config
}
${
extra_train_params
}
"
cmd
=
"
${
python
}
-m paddle.distributed.launch --gpus=
${
gpu
}
${
run_train
}
${
set_use_gpu
}
${
set_save_model
}
${
set_epoch
}
${
set_pretrain
}
${
set_autocast
}
${
set_batchsize
}
${
set_train_params1
}
${
set_amp_config
}
"
else
# train with multi-machine
cmd
=
"
${
python
}
-m paddle.distributed.launch --ips=
${
ips
}
--gpus=
${
gpu
}
${
run_train
}
${
set_use_gpu
}
${
set_save_model
}
${
set_pretrain
}
${
set_epoch
}
${
set_autocast
}
${
set_batchsize
}
${
set_train_params1
}
${
set_amp_config
}
${
extra_train_params
}
"
cmd
=
"
${
python
}
-m paddle.distributed.launch --ips=
${
ips
}
--gpus=
${
gpu
}
${
run_train
}
${
set_use_gpu
}
${
set_save_model
}
${
set_pretrain
}
${
set_epoch
}
${
set_autocast
}
${
set_batchsize
}
${
set_train_params1
}
${
set_amp_config
}
"
fi
# run train
eval
$cmd
...
...
tools/program.py
浏览文件 @
19c5ab23
...
...
@@ -46,8 +46,8 @@ class ArgsParser(ArgumentParser):
self
.
add_argument
(
'-p'
,
'--profiler_options'
,
type
=
str
,
default
=
Non
e
,
type
=
bool
,
default
=
Fals
e
,
help
=
'The option of profiler, which should be in format
\"
key1=value1;key2=value2;key3=value3
\"
.'
)
...
...
@@ -150,6 +150,10 @@ def train(config,
print_batch_step
=
config
[
'Global'
][
'print_batch_step'
]
eval_batch_step
=
config
[
'Global'
][
'eval_batch_step'
]
profiler_options
=
config
[
'profiler_options'
]
if
profiler_options
is
True
:
profiler_options
=
"batch_range=[10,20];state=GPU;tracer_option=Default;profile_path=model.profile"
else
:
profiler_options
=
None
global_step
=
0
if
'global_step'
in
pre_best_model_dict
:
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录