Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
PaddlePaddle
PaddleOCR
提交
49ecf9c3
P
PaddleOCR
项目概览
PaddlePaddle
/
PaddleOCR
大约 1 年 前同步成功
通知
1528
Star
32962
Fork
6643
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
108
列表
看板
标记
里程碑
合并请求
7
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
P
PaddleOCR
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
108
Issue
108
列表
看板
标记
里程碑
合并请求
7
合并请求
7
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
提交
49ecf9c3
编写于
2月 23, 2022
作者:
Z
zhangyikun02
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
add use_xpu config for det_mv3_db.yml
上级
d6ec303e
变更
2
隐藏空白更改
内联
并排
Showing
2 changed file
with
31 addition
and
1 deletion
+31
-1
configs/det/det_mv3_db.yml
configs/det/det_mv3_db.yml
+1
-0
tools/program.py
tools/program.py
+30
-1
未找到文件。
configs/det/det_mv3_db.yml
浏览文件 @
49ecf9c3
Global
:
use_gpu
:
true
use_xpu
:
false
epoch_num
:
1200
log_smooth_window
:
20
print_batch_step
:
10
...
...
tools/program.py
浏览文件 @
49ecf9c3
...
...
@@ -130,6 +130,25 @@ def check_gpu(use_gpu):
pass
def
check_xpu
(
use_xpu
):
"""
Log error and exit when set use_xpu=true in paddlepaddle
cpu/gpu version.
"""
err
=
"Config use_xpu cannot be set as true while you are "
\
"using paddlepaddle cpu/gpu version !
\n
Please try:
\n
"
\
"
\t
1. Install paddlepaddle-xpu to run model on XPU
\n
"
\
"
\t
2. Set use_xpu as false in config file to run "
\
"model on CPU/GPU"
try
:
if
use_xpu
and
not
paddle
.
is_compiled_with_xpu
():
print
(
err
)
sys
.
exit
(
1
)
except
Exception
as
e
:
pass
def
train
(
config
,
train_dataloader
,
valid_dataloader
,
...
...
@@ -512,6 +531,12 @@ def preprocess(is_train=False):
use_gpu
=
config
[
'Global'
][
'use_gpu'
]
check_gpu
(
use_gpu
)
# check if set use_xpu=True in paddlepaddle cpu/gpu version
use_xpu
=
False
if
'use_xpu'
in
config
[
'Global'
]:
use_xpu
=
config
[
'Global'
][
'use_xpu'
]
check_xpu
(
use_xpu
)
alg
=
config
[
'Architecture'
][
'algorithm'
]
assert
alg
in
[
'EAST'
,
'DB'
,
'SAST'
,
'Rosetta'
,
'CRNN'
,
'STARNet'
,
'RARE'
,
'SRN'
,
...
...
@@ -519,7 +544,11 @@ def preprocess(is_train=False):
'SEED'
,
'SDMGR'
,
'LayoutXLM'
,
'LayoutLM'
]
device
=
'gpu:{}'
.
format
(
dist
.
ParallelEnv
().
dev_id
)
if
use_gpu
else
'cpu'
device
=
'cpu'
if
use_gpu
:
device
=
'gpu:{}'
.
format
(
dist
.
ParallelEnv
().
dev_id
)
if
use_xpu
:
device
=
'xpu'
device
=
paddle
.
set_device
(
device
)
config
[
'Global'
][
'distributed'
]
=
dist
.
get_world_size
()
!=
1
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录