|
|
@@ -377,7 +377,7 @@ class WarmUpLR(LearningRateSchedule): |
|
|
Examples: |
|
|
Examples: |
|
|
>>> learning_rate = 0.1 |
|
|
>>> learning_rate = 0.1 |
|
|
>>> warmup_steps = 2 |
|
|
>>> warmup_steps = 2 |
|
|
>>> global_step = Tenosr(2, mstype.int32) |
|
|
|
|
|
|
|
|
>>> global_step = Tensor(2, mstype.int32) |
|
|
>>> warmup_lr = WarmUpLR(learning_rate, warmup_steps) |
|
|
>>> warmup_lr = WarmUpLR(learning_rate, warmup_steps) |
|
|
>>> warmup_lr(global_step) |
|
|
>>> warmup_lr(global_step) |
|
|
""" |
|
|
""" |
|
|
|