Browse Source

reorder fused optimizer

tags/v1.0.0
wilfChen 5 years ago
parent
commit
13dd31f56c
1 changed files with 4 additions and 0 deletions
  1. +4
    -0
      mindspore/ccsrc/utils/utils.h

+ 4
- 0
mindspore/ccsrc/utils/utils.h View File

@@ -344,6 +344,10 @@ const std::set<std::string> kOptOperatorSet = {
kApplyProximalAdagradOpName,
kApplyProximalGradientDescentOpName,
kApplyRMSPropOpName,
kFusedAdamWeightDecayName,
kFusedAdamName,
kFusedWeightScaleApplyMomentum,
kFusedScaleApplyMomentum,
kPullOpName,
};



Loading…
Cancel
Save