optim.scheduler package

Submodules

optim.scheduler.base_scheduler module

class optim.scheduler.base_scheduler.BaseLRScheduler(opts)

Bases: object

__init__(opts) → None

classmethod add_arguments(parser: ArgumentParser) → ArgumentParser

get_lr(epoch: int, curr_iter: int)

update_lr(optimizer, epoch: int, curr_iter: int)

static retrieve_lr(optimizer) → list

extra_repr() → str

Extra information to be represented in __repr__. Each line in the output string should be prefixed with `` ``.

optim.scheduler.cosine module

class optim.scheduler.cosine.CosineScheduler(opts, **kwargs)

Bases: BaseLRScheduler

Cosine learning rate scheduler: https://arxiv.org/abs/1608.03983

__init__(opts, **kwargs) → None

classmethod add_arguments(parser: ArgumentParser) → ArgumentParser

get_lr(epoch: int, curr_iter: int) → float

optim.scheduler.cyclic module

class optim.scheduler.cyclic.CyclicLRScheduler(opts, **kwargs)

Bases: BaseLRScheduler

Cyclic LR: https://arxiv.org/abs/1811.11431

__init__(opts, **kwargs) → None

classmethod add_arguments(parser: ArgumentParser) → ArgumentParser

get_lr(epoch: int, curr_iter: int) → float

optim.scheduler.fixed module

class optim.scheduler.fixed.FixedLRScheduler(opts, **kwargs)

Bases: BaseLRScheduler

Fixed learning rate scheduler with optional linear warm-up strategy

__init__(opts, **kwargs) → None

classmethod add_arguments(parser: ArgumentParser) → ArgumentParser

get_lr(epoch: int, curr_iter: int) → float

optim.scheduler.multi_step module

class optim.scheduler.multi_step.MultiStepLRScheduler(opts, **kwargs)

Bases: BaseLRScheduler

Multi-step learning rate scheduler with optional linear warm-up strategy

__init__(opts, **kwargs) → None

classmethod add_arguments(parser: ArgumentParser) → ArgumentParser

get_lr(epoch: int, curr_iter: int) → float

optim.scheduler.polynomial module

class optim.scheduler.polynomial.PolynomialScheduler(opts, **kwargs)

Bases: BaseLRScheduler

Polynomial LR scheduler

__init__(opts, **kwargs) → None

classmethod add_arguments(parser: ArgumentParser) → ArgumentParser

get_lr(epoch: int, curr_iter: int) → float

Module contents

optim.scheduler.build_scheduler(opts: Namespace, *args, **kwargs) → BaseLRScheduler

optim.scheduler.general_lr_sch_args(parser: ArgumentParser) → ArgumentParser

optim.scheduler.arguments_scheduler(parser: ArgumentParser) → ArgumentParser