ls_mlkit.scheduler package

Submodules

• ls_mlkit.scheduler.lr_scheduler_factory module

Module contents

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