ls_mlkit.scheduler.lr_scheduler_factory module

ls_mlkit.scheduler.lr_scheduler_factory.get_lambda_lr_scheduler(optimizer, num_warmup_steps, num_training_steps, lr_scheduler_type='linear')

ls_mlkit.scheduler.lr_scheduler_factory.get_lr_scheduler(optimizer, n_warmup_steps, n_training_steps, lr_scheduler_type: Literal['cosine', 'linear', 'constant', 'cosine_annealing', 'cosine_with_warmup'] = 'linear')