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')[source]
ls_mlkit.scheduler.lr_scheduler_factory.get_lr_scheduler(optimizer, n_warmup_steps, n_training_steps, lr_scheduler_type='linear')[source]