p60975123
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!531 Check parameter power of polynomial_decay_lr 

Merge pull request !531 from fanglei/checkpower
This commit is contained in:
mindspore-ci-bot 2020-04-21 20:03:13 +08:00 committed by Gitee
parent 1fa06c6045 b87643958f
commit cbdc23d777
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mindspore/nn/dynamic_lr.py
View File

@ -269,7 +269,7 @@ def polynomial_decay_lr(learning_rate, end_learning_rate, total_step, step_per_e
total_step (int): The total number of steps. total_step (int): The total number of steps.
step_per_epoch (int): The number of steps in per epoch. step_per_epoch (int): The number of steps in per epoch.
decay_epoch (int): A value used to calculate decayed learning rate. decay_epoch (int): A value used to calculate decayed learning rate.
power (float): A value used to calculate decayed learning rate. power (float): A value used to calculate decayed learning rate. This parameter should be greater than 0.
update_decay_epoch (bool): If true, update `decay_epoch`. Default: False. update_decay_epoch (bool): If true, update `decay_epoch`. Default: False.
Returns: Returns: