Skip to content

Commit 30a877b

Browse files
Update torch/optim/lr_scheduler.py
Co-authored-by: Jane (Yuan) Xu <31798555+janeyx99@users.noreply.github.com>
1 parent daa7f99 commit 30a877b

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

torch/optim/lr_scheduler.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -1390,7 +1390,7 @@ def load_state_dict(self, state_dict):
13901390
class CosineAnnealingWarmRestarts(LRScheduler):
13911391
r"""Set the learning rate of each parameter group using a cosine annealing schedule.
13921392
1393-
Where :math:`\eta_{max}` is set to the initial lr, :math:`T_{cur}`
1393+
:math:`\eta_{max}` is set to the initial lr, :math:`T_{cur}`
13941394
is the number of epochs since the last restart and :math:`T_{i}` is the number
13951395
of epochs between two warm restarts in SGDR:
13961396

0 commit comments

Comments
 (0)