diff --git a/torch/optim/lr_scheduler.py b/torch/optim/lr_scheduler.py index 0df5b53283fb8..752e8606aa6a5 100644 --- a/torch/optim/lr_scheduler.py +++ b/torch/optim/lr_scheduler.py @@ -83,7 +83,7 @@ class StepLR(_LRScheduler): last_epoch (int): The index of last epoch. Default: -1. Example: - >>> # Assuming optimizer uses lr = 0.5 for all groups + >>> # Assuming optimizer uses lr = 0.05 for all groups >>> # lr = 0.05 if epoch < 30 >>> # lr = 0.005 if 30 <= epoch < 60 >>> # lr = 0.0005 if 60 <= epoch < 90 @@ -118,7 +118,7 @@ class MultiStepLR(_LRScheduler): last_epoch (int): The index of last epoch. Default: -1. Example: - >>> # Assuming optimizer uses lr = 0.5 for all groups + >>> # Assuming optimizer uses lr = 0.05 for all groups >>> # lr = 0.05 if epoch < 30 >>> # lr = 0.005 if 30 <= epoch < 80 >>> # lr = 0.0005 if epoch >= 80