mmagic.engine.schedulers.linear_lr_scheduler_with_interval

Module Contents

Classes

LinearLrInterval

Linear learning rate scheduler for image generation.

class mmagic.engine.schedulers.linear_lr_scheduler_with_interval.LinearLrInterval(*args, interval=1, **kwargs)

Bases: mmengine.optim.LinearLR

Linear learning rate scheduler for image generation.

In the beginning, the learning rate is ‘start_factor’ defined in mmengine. We give a target learning rate ‘end_factor’ and a start point ‘begin’. If :attr:self.by_epoch is True, ‘begin’ is calculated by epoch, otherwise, calculated by iteration.” Before ‘begin’, we fix learning rate as ‘start_factor’; After ‘begin’, we linearly update learning rate to ‘end_factor’.

Parameters

interval (int) – The interval to update the learning rate. Default: 1.

_get_value()

Compute value using chainable form of the scheduler.