Browse Source

Parameter power of polynomial_decay_lr should be greater than 0

tags/v0.2.0-alpha
leilei_snow 5 years ago
parent
commit
b87643958f
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