nntoolbox.callbacks.lr_scheduler module

class nntoolbox.callbacks.lr_scheduler.LRSchedulerCB(scheduler, timescale: str = 'iter')

Bases: nntoolbox.callbacks.callbacks.Callback

on_batch_end(logs: Dict[str, Any])

on_epoch_end(logs: Dict[str, Any]) → bool

class nntoolbox.callbacks.lr_scheduler.ReduceLROnPlateauCB(optimizer: torch.optim.optimizer.Optimizer, monitor: str = 'accuracy', mode: str = 'max', factor: float = 0.1, patience: int = 10, verbose: bool = True, threshold: float = 0.0001, threshold_mode: str = 'rel', cooldown: int = 0, min_lr: float = 0, eps: float = 1e-08)

Bases: nntoolbox.callbacks.callbacks.Callback

on_epoch_end(logs: Dict[str, Any]) → bool