Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
magicwindyyd
mindspore
提交
40a3a714
M
mindspore
项目概览
magicwindyyd
/
mindspore
与 Fork 源项目一致
Fork自
MindSpore / mindspore
通知
1
Star
1
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
0
列表
看板
标记
里程碑
合并请求
0
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
M
mindspore
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
0
Issue
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
提交
40a3a714
编写于
4年前
作者:
M
mindspore-ci-bot
提交者:
Gitee
4年前
浏览文件
操作
浏览文件
下载
差异文件
!5148 change group conv dtype in gpu resnext50
Merge pull request !5148 from zhaoting/master
上级
b5ed5466
9ef6e72c
master
无相关合并请求
变更
2
隐藏空白更改
内联
并排
Showing
2 changed file
with
2 addition
and
6 deletion
+2
-6
model_zoo/official/cv/resnext50/src/utils/auto_mixed_precision.py
...o/official/cv/resnext50/src/utils/auto_mixed_precision.py
+0
-3
model_zoo/official/cv/resnext50/train.py
model_zoo/official/cv/resnext50/train.py
+2
-3
未找到文件。
model_zoo/official/cv/resnext50/src/utils/auto_mixed_precision.py
浏览文件 @
40a3a714
...
...
@@ -44,9 +44,6 @@ def auto_mixed_precision(network):
elif
name
==
'fc'
:
network
.
insert_child_to_cell
(
name
,
OutputTo
(
subcell
,
mstype
.
float32
))
change
=
True
elif
name
==
'conv2'
:
subcell
.
to_float
(
mstype
.
float32
)
change
=
True
elif
isinstance
(
subcell
,
(
nn
.
BatchNorm2d
,
nn
.
BatchNorm1d
)):
network
.
insert_child_to_cell
(
name
,
OutputTo
(
subcell
.
to_float
(
mstype
.
float32
),
mstype
.
float16
))
change
=
True
...
...
This diff is collapsed.
Click to expand it.
model_zoo/official/cv/resnext50/train.py
浏览文件 @
40a3a714
...
...
@@ -36,7 +36,6 @@ from src.warmup_cosine_annealing_lr import warmup_cosine_annealing_lr
from
src.utils.logging
import
get_logger
from
src.utils.optimizers__init__
import
get_param_groups
from
src.image_classification
import
get_network
from
src.utils.auto_mixed_precision
import
auto_mixed_precision
from
src.config
import
config
...
...
@@ -273,8 +272,8 @@ def train(cloud_args=None):
model
=
Model
(
network
,
loss_fn
=
loss
,
optimizer
=
opt
,
loss_scale_manager
=
loss_scale_manager
,
metrics
=
{
'acc'
},
amp_level
=
"O3"
)
else
:
auto_mixed_precision
(
network
)
model
=
Model
(
network
,
loss_fn
=
loss
,
optimizer
=
opt
,
loss_scale_manager
=
loss_scale_manager
,
metrics
=
{
'acc'
}
)
model
=
Model
(
network
,
loss_fn
=
loss
,
optimizer
=
opt
,
loss_scale_manager
=
loss_scale_manager
,
metrics
=
{
'acc'
},
amp_level
=
"O2"
)
# checkpoint save
progress_cb
=
ProgressMonitor
(
args
)
...
...
This diff is collapsed.
Click to expand it.
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录
反馈
建议
客服
返回
顶部