construct_scheduler#

construct_scheduler(optimizer, scheduler_cfg)#

Creates a learning rate scheduler for a given optimizer given a configuration.

Parameters:
  • optimizer – the optimizer to be used

  • scheduler_cfg (SchedulerConfig) – The scheduler configuration.

Returns:

The constructed scheduler.

Return type:

torch.optim.lr_scheduler.LRScheduler