espnet3.components.optimizers.multiple_scheduler.MultipleScheduler
Less than 1 minute
espnet3.components.optimizers.multiple_scheduler.MultipleScheduler
class espnet3.components.optimizers.multiple_scheduler.MultipleScheduler(multiple_optimizer: MultipleOptimizer, lr_scheduler: LRScheduler, optim_idx: int)
Bases: _LRScheduler
Wrapper class around
``lr_scheduler``s to return a dummy optimizer.
Modified from the reply in a GitHub Issue thread here: https://github.com/Lightning-AI/lightning/issues/3346#issuecomment-1036063687
- Parameters:
- multiple_optim – MultipleOptimizer
- lr_scheduler – torch.optim.lr_scheduler._LRScheduler
- optim_idx – int Index of the optimizer in
multiple_optimthe learning rate schedulerlr_scheduleris assigned to
Initialize MultipleScheduler object.
