ls_mlkit.util.scheduler module

class ls_mlkit.util.scheduler.ObjectAttrScheduler(obj, attr_name, total=None, warmup_steps=None, warmup_ratio=0, strategy='constant')

Bases: object

get()

step()

class ls_mlkit.util.scheduler.ObjectAttrsScheduler(obj, attr_names, total=None, warmup_steps=None, warmup_ratio=0, strategy='constant')

Bases: object

get()

step()

class ls_mlkit.util.scheduler.Scheduler(info, total, warmup_steps=None, warmup_ratio=0)

Bases: object

get(key=None)

step()

ls_mlkit.util.scheduler.constant(value, current, total, warmup_steps=0)

ls_mlkit.util.scheduler.cosine_decay_with_warmup(value, current, total, warmup_steps=0)

ls_mlkit.util.scheduler.linear_decay_with_warmup(value, current, total, warmup_steps=0)