FastSurferCNN.utils.lr_scheduler¶

FastSurferCNN.utils.lr_scheduler.get_lr_scheduler(optimizer, cfg)[source]¶

Give a schedular for left-right scheduling.

Parameters:

optimizertorch.optim.Optimizer: Optimizer for the scheduler.
cfgyacs.config.CfgNode: Configuration node.

Returns:

Union[None, scheduler.StepLR, scheduler.CosineAnnealingWarmRestarts]: A learning rate scheduler configured according to cfg, or None if no scheduling is required.

Raises:

ValueError: lr scheduler is not supported.