diff --git a/mindspore/ccsrc/backend/kernel_compiler/gpu/nn/fused_adam_weight_decay.h b/mindspore/ccsrc/backend/kernel_compiler/gpu/nn/fused_adam_weight_decay.h index 10e90aef8d..9a345682cd 100644 --- a/mindspore/ccsrc/backend/kernel_compiler/gpu/nn/fused_adam_weight_decay.h +++ b/mindspore/ccsrc/backend/kernel_compiler/gpu/nn/fused_adam_weight_decay.h @@ -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; }