Skip to content

M
mindspore

正统之独孤求败 / mindspore
与 Fork 源项目一致

Fork自 MindSpore / mindspore

通知 1
Star 0
Fork 0
M
mindspore

体验新版 GitCode，发现更多精彩内容 >>

提交 bd2fd31a 编写于 作者: S simson
浏览文件
操作

revert limitation of end_learning_rate

上级 88215d00
隐藏空白更改
内联 并排
Showing with 1 addition and 1 deletion
mindspore/nn/optim/lamb.py
浏览文件 @ bd2fd31a
...@@ -114,7 +114,7 @@ def _check_param_value(decay_steps, warmup_steps, start_learning_rate, ...@@ -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)
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或