forked from mindspore-Ecosystem/mindspore
!531 Check parameter power of polynomial_decay_lr
Merge pull request !531 from fanglei/checkpower
This commit is contained in:
commit
cbdc23d777
|
@ -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.
|
||||
step_per_epoch (int): The number of steps in per epoch.
|
||||
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.
|
||||
|
||||
Returns:
|
||||
|
|
Loading…
Reference in New Issue