Fix StepLR docs (#4478)

zou3519 · soumith · commit 3321cdce84a4 · 2018-02-06T21:41:03.000-08:00
diff --git a/torch/optim/lr_scheduler.py b/torch/optim/lr_scheduler.py
@@ -82,7 +82,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
@@ -117,7 +117,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
