p04896132
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

update piecewise_constant_lr support tuple input

This commit is contained in:
leilei_snow 2020-04-21 02:50:38 +00:00
parent 6b0ff88b1c
commit 742395da12
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
mindspore/nn/dynamic_lr.py
View File

@ -31,8 +31,8 @@ def piecewise_constant_lr(milestone, learning_rates):
y[i] = x_t for i \in [M_{t-1}, M_t)
Args:
milestone (list[int]): A list of milestone. This list is a monotone increasing list.
learning_rates (list[float]): A list of learning rates.
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.
Returns:
list[float]. The size of list is :math:`M_N`.