Generated with

docs/torch/torch.optim/torch.optim.lr_scheduler/StepLR

StepLR

torch.optim.lr_scheduler.StepLR

class StepLR(optimizer: Optimizer, step_size: Int, gamma: Float) extends LRScheduler

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
Graph
Supertypes: trait LRScheduler

class Object

trait Matchable

class Any

Members list

Value members

Concrete methods

Attributes

Source: StepLR.scala

In this article

Generated with