Browse Source

!531 Check parameter power of polynomial_decay_lr

Merge pull request !531 from fanglei/checkpower
tags/v0.2.0-alpha
mindspore-ci-bot Gitee 6 years ago
parent
commit
cbdc23d777
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      mindspore/nn/dynamic_lr.py

+ 1
- 1
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.
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…
Cancel
Save