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
!8173
fix gpu adam kernel fusion
Merge pull request
!8173
from chenweifeng/fix-gpu-adam-fusion
tags/v1.1.0
mindspore-ci-bot
Gitee
5 years ago
parent
57fc8f45ae
3f472ce669
commit
cf64fe4e07
1 changed files
with
1 additions
and
1 deletions
Split View
Diff Options
Show Stats
Download Patch File
Download Diff File
+1
-1
mindspore/ccsrc/backend/kernel_compiler/gpu/nn/fused_adam_weight_decay.h
+ 1
- 1
mindspore/ccsrc/backend/kernel_compiler/gpu/nn/fused_adam_weight_decay.h
View File
@@ -33,7 +33,7 @@ class FusedAdamWeightDecayGpuKernel : public GpuKernel {
bool Init(const CNodePtr &kernel_node) override {
auto node_name = AnfAlgo::GetCNodeName(kernel_node);
if (node_name == "AdamWeighDecay") {
if (node_name == "
Fused
AdamWeigh
t
Decay") {
weight_decay_ = true;
}
Write
Preview
Loading…
Cancel
Save