This website works better with JavaScript.
Home
Issues
Pull Requests
Milestones
AI流水线
Repositories
Datasets
Forum
实训
竞赛
大数据
AI开发
Register
Sign In
Huawei_Technology
/
mindspore
Not watched
Unwatch
Watch all
Watch but not notify
1
Star
0
Fork
0
Code
Releases
13
Wiki
evaluate
Activity
Issues
0
Pull Requests
0
Datasets
Model
Cloudbrain
HPC
Browse Source
!10098
fix bug of pynative's mixprecision
From:
@lianliguang
Reviewed-by: @kisnwang,@zhoufeng54 Signed-off-by:
@zhoufeng54
tags/v1.1.0
mindspore-ci-bot
Gitee
5 years ago
parent
c2579d1dc6
168fd6c9b2
commit
885f41549a
1 changed files
with
1 additions
and
1 deletions
Split View
Diff Options
Show Stats
Download Patch File
Download Diff File
+1
-1
mindspore/nn/cell.py
+ 1
- 1
mindspore/nn/cell.py
View File
@@ -647,7 +647,7 @@ class Cell(Cell_):
param.set_cast_dtype(mstype.float32)
elif self._mindspore_flags.get('fp16'):
param.set_cast_dtype(mstype.float16)
else:
el
if hasattr(param, "
se
t_cast_dtype")
:
# retest dtype
param.set_cast_dtype()
return param
Write
Preview
Loading…
Cancel
Save