GenericScheduler

class lightning_ir.schedulers.schedulers.GenericScheduler(*args, keys: Sequence[str] | None = None, **kwargs)

Bases: Callback, ABC

__init__(*args, keys: Sequence[str] | None = None, **kwargs) → None

Methods

__init__(*args[, keys])
get_value(sub_keys, obj)
on_train_batch_start(trainer, pl_module, ...)	Called when the train batch begins.
on_train_end(trainer, pl_module)	Called when the train ends.
on_train_start(trainer, pl_module)	Called when the train begins.
set_value(sub_keys, obj, value)
step(key, current_step)
value_lambda(current_step)

Attributes

on_train_batch_start(trainer: Trainer, pl_module: LightningModule, batch: Any, batch_idx: int) → None

Called when the train batch begins.

on_train_end(trainer: Trainer, pl_module: LightningModule) → None

Called when the train ends.

on_train_start(trainer: Trainer, pl_module: LightningIRModule) → None

Called when the train begins.