Class LRScheduler

• Defined in File lr_scheduler.h

Inheritance Relationships

Derived Type

• public torch::optim::StepLR (Class StepLR)

Class Documentation

class LRScheduler

Subclassed by torch::optim::StepLR

Public Functions

LRScheduler(torch::optim::Optimizer &optimizer)

virtual ~LRScheduler() = default

void step()

Protected Functions

virtual std::vector<double> get_lrs() = 0

std::vector<double> get_current_lrs() const

Protected Attributes

unsigned step_count_ = {}