Browse Source

fix gpu adam fusion kernel

tags/v1.1.0
wilfChen 5 years ago
parent
commit
3f472ce669
1 changed files with 1 additions and 1 deletions
  1. +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 == "FusedAdamWeightDecay") {
weight_decay_ = true;
}



Loading…
Cancel
Save