Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
PaddlePaddle
PaddleClas
提交
efde56ff
P
PaddleClas
项目概览
PaddlePaddle
/
PaddleClas
大约 1 年 前同步成功
通知
115
Star
4999
Fork
1114
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
19
列表
看板
标记
里程碑
合并请求
6
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
P
PaddleClas
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
19
Issue
19
列表
看板
标记
里程碑
合并请求
6
合并请求
6
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
未验证
提交
efde56ff
编写于
4月 10, 2022
作者:
G
gaotingquan
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
fix: only fp16 evaluation is supported when ampO2 is enabled
上级
b761325f
变更
3
隐藏空白更改
内联
并排
Showing
3 changed file
with
5 addition
and
1 deletion
+5
-1
ppcls/configs/ImageNet/ResNet/ResNet50_amp_O2_ultra.yaml
ppcls/configs/ImageNet/ResNet/ResNet50_amp_O2_ultra.yaml
+2
-0
ppcls/configs/ImageNet/SENet/SE_ResNeXt101_32x4d_amp_O2_ultra.yaml
...figs/ImageNet/SENet/SE_ResNeXt101_32x4d_amp_O2_ultra.yaml
+2
-0
ppcls/engine/evaluation/classification.py
ppcls/engine/evaluation/classification.py
+1
-1
未找到文件。
ppcls/configs/ImageNet/ResNet/ResNet50_amp_O2_ultra.yaml
浏览文件 @
efde56ff
...
...
@@ -24,6 +24,8 @@ AMP:
use_dynamic_loss_scaling
:
True
# O2: pure fp16
level
:
O2
# only FP16 evaluation is supported when AMP O2 is enabled
use_fp16_test
:
True
# model architecture
Arch
:
...
...
ppcls/configs/ImageNet/SENet/SE_ResNeXt101_32x4d_amp_O2_ultra.yaml
浏览文件 @
efde56ff
...
...
@@ -38,6 +38,8 @@ AMP:
use_dynamic_loss_scaling
:
True
# O2: pure fp16
level
:
O2
# only FP16 evaluation is supported when AMP O2 is enabled
use_fp16_test
:
True
Optimizer
:
name
:
Momentum
...
...
ppcls/engine/evaluation/classification.py
浏览文件 @
efde56ff
...
...
@@ -53,7 +53,7 @@ def classification_eval(engine, epoch_id=0):
]
time_info
[
"reader_cost"
].
update
(
time
.
time
()
-
tic
)
batch_size
=
batch
[
0
].
shape
[
0
]
batch
[
0
]
=
paddle
.
to_tensor
(
batch
[
0
])
.
astype
(
"float32"
)
batch
[
0
]
=
paddle
.
to_tensor
(
batch
[
0
])
if
not
engine
.
config
[
"Global"
].
get
(
"use_multilabel"
,
False
):
batch
[
1
]
=
batch
[
1
].
reshape
([
-
1
,
1
]).
astype
(
"int64"
)
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录