towhee.trainer.utils.plot_utils.plot_lrs_for_config

towhee.trainer.utils.plot_utils.plot_lrs_for_config(configs: TrainingConfig, num_training_steps: int, start_lr: int = 100)

Plot the lr for a config. It’s useful to visualize how the lr scheduler control the lr by the yaml file configs. :param configs: The training config, which can be load from the yaml file. :type configs: TrainingConfig :param num_training_steps: Total training step number. :type num_training_steps: int :param start_lr: The start lr value. :type start_lr: int