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]