Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
PaddlePaddle
PaddleOCR
提交
b0e3c3f2
P
PaddleOCR
项目概览
PaddlePaddle
/
PaddleOCR
1 年多 前同步成功
通知
1541
Star
32964
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看板
未验证
提交
b0e3c3f2
编写于
7月 10, 2023
作者:
X
xiaoting
提交者:
GitHub
7月 10, 2023
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
clear fluid (#10333)
上级
722e9dc0
变更
5
隐藏空白更改
内联
并排
Showing
5 changed file
with
4 addition
and
6 deletion
+4
-6
ppocr/optimizer/optimizer.py
ppocr/optimizer/optimizer.py
+1
-2
test_tipc/supplementary/data_loader.py
test_tipc/supplementary/data_loader.py
+0
-1
test_tipc/supplementary/train.py
test_tipc/supplementary/train.py
+1
-1
tools/eval.py
tools/eval.py
+1
-1
tools/infer/utility.py
tools/infer/utility.py
+1
-1
未找到文件。
ppocr/optimizer/optimizer.py
浏览文件 @
b0e3c3f2
...
...
@@ -84,8 +84,7 @@ class Adam(object):
if
self
.
group_lr
:
if
self
.
training_step
==
'LF_2'
:
import
paddle
if
isinstance
(
model
,
paddle
.
fluid
.
dygraph
.
parallel
.
DataParallel
):
# multi gpu
if
isinstance
(
model
,
paddle
.
DataParallel
):
# multi gpu
mlm
=
model
.
_layers
.
head
.
MLM_VRM
.
MLM
.
parameters
()
pre_mlm_pp
=
model
.
_layers
.
head
.
MLM_VRM
.
Prediction
.
pp_share
.
parameters
(
)
...
...
test_tipc/supplementary/data_loader.py
浏览文件 @
b0e3c3f2
import
numpy
as
np
from
paddle.vision.datasets
import
Cifar100
from
paddle.vision.transforms
import
Normalize
from
paddle.fluid.dataloader.collate
import
default_collate_fn
import
signal
import
os
from
paddle.io
import
Dataset
,
DataLoader
,
DistributedBatchSampler
...
...
test_tipc/supplementary/train.py
浏览文件 @
b0e3c3f2
...
...
@@ -71,7 +71,7 @@ def amp_scaler(config):
'FLAGS_cudnn_batchnorm_spatial_persistent'
:
1
,
'FLAGS_max_inplace_grad_add'
:
8
,
}
paddle
.
fluid
.
set_flags
(
AMP_RELATED_FLAGS_SETTING
)
paddle
.
set_flags
(
AMP_RELATED_FLAGS_SETTING
)
scale_loss
=
config
[
"AMP"
].
get
(
"scale_loss"
,
1.0
)
use_dynamic_loss_scaling
=
config
[
"AMP"
].
get
(
"use_dynamic_loss_scaling"
,
False
)
...
...
tools/eval.py
浏览文件 @
b0e3c3f2
...
...
@@ -112,7 +112,7 @@ def main():
'FLAGS_cudnn_batchnorm_spatial_persistent'
:
1
,
'FLAGS_max_inplace_grad_add'
:
8
,
}
paddle
.
fluid
.
set_flags
(
AMP_RELATED_FLAGS_SETTING
)
paddle
.
set_flags
(
AMP_RELATED_FLAGS_SETTING
)
scale_loss
=
config
[
"Global"
].
get
(
"scale_loss"
,
1.0
)
use_dynamic_loss_scaling
=
config
[
"Global"
].
get
(
"use_dynamic_loss_scaling"
,
False
)
...
...
tools/infer/utility.py
浏览文件 @
b0e3c3f2
...
...
@@ -313,7 +313,7 @@ def get_infer_gpuid():
if
sysstr
==
"Windows"
:
return
0
if
not
paddle
.
fluid
.
core
.
is_compiled_with_rocm
()
:
if
not
paddle
.
device
.
is_compiled_with_rocm
:
cmd
=
"env | grep CUDA_VISIBLE_DEVICES"
else
:
cmd
=
"env | grep HIP_VISIBLE_DEVICES"
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录