torch.optim.lr_scheduler
package torch.optim.lr_scheduler
Members list
Type members
Classlikes
trait LRScheduler
Attributes
- Source
- LRScheduler.scala
- Supertypes
-
class Objecttrait Matchableclass Any
- Known subtypes
-
class StepLR
Decays the learning rate of each parameter group by gamma every step_size epochs.
Decays the learning rate of each parameter group by gamma every step_size epochs.
Notice that such decay can happen simultaneously with other changes to the learning rate from outside this scheduler.
Attributes
- Source
- StepLR.scala
- Supertypes
In this article