Browse Source

XXXLR not support cpu: ExponentialDecayLR, PolynomialDecayLR, etc.

pull/14000/head
liuhe 4 years ago
parent
commit
05df8a9b26
2 changed files with 5 additions and 5 deletions
  1. +4
    -4
      mindspore/nn/learning_rate_schedule.py
  2. +1
    -1
      mindspore/ops/operations/array_ops.py

+ 4
- 4
mindspore/nn/learning_rate_schedule.py View File

@@ -280,8 +280,8 @@ class CosineDecayLR(LearningRateSchedule):
ValueError: If `min_lr` is less than 0 or `decay_steps` is less than 1.
ValueError: If `max_lr` is less than or equal to 0.

Supported Platforms:
``Ascend`` ``GPU`` ``CPU``
Supported Platforms:
``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


+ 1
- 1
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)


Loading…
Cancel
Save