Modify wrong type description.

c2129b91 · leilei_snow · 742395da · c2129b91
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

mindspore/nn/dynamic_lr.py mindspore/nn/dynamic_lr.py +1 -1

未找到文件。
--- a/mindspore/nn/dynamic_lr.py
+++ b/mindspore/nn/dynamic_lr.py
@@ -32,7 +32,7 @@ def piecewise_constant_lr(milestone, learning_rates):

    Args:
        milestone (Union[list[int], tuple[int]]): A list of milestone. This list is a monotone increasing list.
-        learning_rates (Union[list[float], tuple[int]]): A list of learning rates.
+        learning_rates (Union[list[float], tuple[float]]): A list of learning rates.

    Returns:
        list[float]. The size of list is :math:`M_N`.