towhee.trainer.utils.plot_utils.plot_lrs_for_scheduler

towhee.trainer.utils.plot_utils.plot_lrs_for_scheduler(optimizer: Optimizer, scheduler: _LRScheduler, total_steps: int)

Plot the lr curve for a specified optimizer and scheduler instance. :param optimizer: The optimizer contains the training parameters of a model. :type optimizer: Optimizer :param scheduler: The lr scheduler for an optimizer. :type scheduler: _LRScheduler :param total_steps: Total training step number. :type total_steps: int