artist.core.learning_rate_schedulers

Functions

`exponential`(→ torch.optim.lr_scheduler.LRScheduler)	Create an exponential learning rate scheduler.
`cyclic`(→ torch.optim.lr_scheduler.LRScheduler)	Create a cyclic learning rate scheduler.
`reduce_on_plateau`(→ torch.optim.lr_scheduler.LRScheduler)	Create learning rate scheduler that reduces on plateaus.

Module Contents

artist.core.learning_rate_schedulers.exponential(optimizer: torch.optim.Optimizer, parameters: dict[str, float]) → torch.optim.lr_scheduler.LRScheduler

Create an exponential learning rate scheduler.

Parameters

optimizerOptimizer: The optimizer.
parametersdict[str, float]: The scheduler parameters.

Returns

LRScheduler: An exponential learning rate scheduler.

artist.core.learning_rate_schedulers.cyclic(optimizer: torch.optim.Optimizer, parameters: dict[str, float]) → torch.optim.lr_scheduler.LRScheduler

Create a cyclic learning rate scheduler.

Parameters

optimizerOptimizer: The optimizer.
parametersdict[str, float]: The scheduler parameters.

Returns

LRScheduler: A cyclic learning rate scheduler.

artist.core.learning_rate_schedulers.reduce_on_plateau(optimizer: torch.optim.Optimizer, parameters: dict[str, float]) → torch.optim.lr_scheduler.LRScheduler

Create learning rate scheduler that reduces on plateaus.

Parameters

optimizerOptimizer: The optimizer.
parametersdict[str, float]: The scheduler parameters.

Returns

LRScheduler: A learning rate scheduler that reduces on plateaus.