From b87643958fa305d86a36b88e1ce20bba910d18a0 Mon Sep 17 00:00:00 2001 From: leilei_snow Date: Tue, 21 Apr 2020 06:52:08 +0000 Subject: [PATCH] Parameter power of polynomial_decay_lr should be greater than 0 --- mindspore/nn/dynamic_lr.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mindspore/nn/dynamic_lr.py b/mindspore/nn/dynamic_lr.py index beed7a0186a..fb4d229a9d3 100644 --- a/mindspore/nn/dynamic_lr.py +++ b/mindspore/nn/dynamic_lr.py @@ -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: