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. ValueError: If `max_lr` is less than or equal to 0.
Supported Platforms: Supported Platforms:
``Ascend`` ``GPU`` ``CPU`` ``Ascend`` ``GPU``
Examples: Examples:
>>> min_lr = 0.01 >>> min_lr = 0.01
@ -357,7 +357,7 @@ class PolynomialDecayLR(LearningRateSchedule):
ValueError: If `learning_rate` or `power` is less than or equal to 0. ValueError: If `learning_rate` or `power` is less than or equal to 0.
Supported Platforms: Supported Platforms:
``Ascend`` ``GPU`` ``CPU`` ``Ascend`` ``GPU``
Examples: Examples:
>>> learning_rate = 0.1 >>> learning_rate = 0.1
@ -436,7 +436,7 @@ class WarmUpLR(LearningRateSchedule):
ValueError: If `learning_rate` is less than or equal to 0. ValueError: If `learning_rate` is less than or equal to 0.
Supported Platforms: Supported Platforms:
``Ascend`` ``GPU`` ``CPU`` ``Ascend`` ``GPU``
Examples: Examples:
>>> learning_rate = 0.1 >>> 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. A 1-D Tensor, with the same type as the inputs.
Supported Platforms: Supported Platforms:
``GPU`` ``GPU`` `CPU`
Examples: Examples:
>>> start = Tensor(0, mstype.int32) >>> start = Tensor(0, mstype.int32)