xxq250
/
mindspore
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

XXXLR not support cpu: ExponentialDecayLR, PolynomialDecayLR, etc.

This commit is contained in:
liuhe 2021-03-24 14:43:54 +08:00
parent b6bf797ae7
commit 05df8a9b26
2 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
mindspore/nn/learning_rate_schedule.py
View File

@ -281,7 +281,7 @@ class CosineDecayLR(LearningRateSchedule):
ValueError: If `max_lr` is less than or equal to 0.
Supported Platforms:
``Ascend`` ``GPU`` ``CPU``
``Ascend`` ``GPU``
Examples:
>>> min_lr = 0.01
@ -357,7 +357,7 @@ class PolynomialDecayLR(LearningRateSchedule):
ValueError: If `learning_rate` or `power` is less than or equal to 0.
Supported Platforms:
``Ascend`` ``GPU`` ``CPU``
``Ascend`` ``GPU``
Examples:
>>> learning_rate = 0.1
@ -436,7 +436,7 @@ class WarmUpLR(LearningRateSchedule):
ValueError: If `learning_rate` is less than or equal to 0.
Supported Platforms:
``Ascend`` ``GPU`` ``CPU``
``Ascend`` ``GPU``
Examples:
>>> learning_rate = 0.1

2
mindspore/ops/operations/array_ops.py
View File

@ -5240,7 +5240,7 @@ class Range(PrimitiveWithCheck):
A 1-D Tensor, with the same type as the inputs.
Supported Platforms:
``GPU``
``GPU`` `CPU`
Examples:
>>> start = Tensor(0, mstype.int32)