Browse Source

!921 Revert limitation of end_learning_rate

Merge pull request !921 from Simson/push-to-opensource
tags/v0.3.0-alpha
mindspore-ci-bot Gitee 5 years ago
parent
commit
2b3192259d
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      mindspore/nn/optim/lamb.py

+ 1
- 1
mindspore/nn/optim/lamb.py View File

@@ -114,7 +114,7 @@ def _check_param_value(decay_steps, warmup_steps, start_learning_rate,
_ = warmup_steps _ = warmup_steps
validator.check_float_positive('start_learning_rate', start_learning_rate, prim_name) validator.check_float_positive('start_learning_rate', start_learning_rate, prim_name)
validator.check_float_legal_value('start_learning_rate', start_learning_rate, prim_name) validator.check_float_legal_value('start_learning_rate', start_learning_rate, prim_name)
validator.check_float_positive('end_learning_rate', end_learning_rate, prim_name)
validator.check_value_type("end_learning_rate", end_learning_rate, [float], prim_name)
validator.check_float_legal_value('end_learning_rate', end_learning_rate, prim_name) validator.check_float_legal_value('end_learning_rate', end_learning_rate, prim_name)
validator.check_float_positive('power', power, prim_name) validator.check_float_positive('power', power, prim_name)
validator.check_float_legal_value('power', power, prim_name) validator.check_float_legal_value('power', power, prim_name)


Loading…
Cancel
Save