commit 07a08fb95f7a0dfa4b721695a44f71aafa0b1730
author hesom <5159852+hesom@users.noreply.github.com> Fri Nov 05 07:08:57 2021 -0700
committer Facebook GitHub Bot <facebook-github-bot@users.noreply.github.com> Fri Nov 05 07:16:15 2021 -0700
tree 71461d84f36393ba161e1babff611836791c2e38
parent 53ebccbe78349ac28ca54f211a5dc0de60b775df

Fix typo in LinearLR docs (#67840)

Summary:
The final learning rate should be 0.05 like the lr used as the argument for the optimizer and not 0.005.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/67840

Reviewed By: jbschlosser

Differential Revision: D32187091

Pulled By: albanD

fbshipit-source-id: 8aff691bba3896a847d7b9d9d669a65f67a6f066

torch/optim/lr_scheduler.py

diff --git a/torch/optim/lr_scheduler.py b/torch/optim/lr_scheduler.py
index b444fe7..16fb199 100644
--- a/torch/optim/lr_scheduler.py
+++ b/torch/optim/lr_scheduler.py
@@ -509,7 +509,7 @@
         >>> # lr = 0.03125  if epoch == 1
         >>> # lr = 0.0375   if epoch == 2
         >>> # lr = 0.04375  if epoch == 3
-        >>> # lr = 0.005    if epoch >= 4
+        >>> # lr = 0.05    if epoch >= 4
         >>> scheduler = LinearLR(self.opt, start_factor=0.5, total_iters=4)
         >>> for epoch in range(100):
         >>>     train(...)