pytorch_pfn_extras.training.extensions.lr_scheduler#

Classes

pytorch_pfn_extras.training.extensions.lr_scheduler.Any(...)

Special type indicating an unconstrained type.

pytorch_pfn_extras.training.extensions.lr_scheduler.ExtensionsManagerProtocol(...)

pytorch_pfn_extras.training.extensions.lr_scheduler.LRScheduler(...)

Trainer extension to adjust the learning rate using PyTorch's learning rate scheduler.

pytorch_pfn_extras.training.extensions.lr_scheduler.ReduceLROnPlateau(...)

Reduce learning rate when a metric has stopped improving.