aiaccel.torch.lightning.OptimizerConfig

class aiaccel.torch.lightning.OptimizerConfig(optimizer_generator: Callable[..., optim.optimizer.Optimizer], scheduler_generator: Callable[..., optim.lr_scheduler.LRScheduler] | None = None, scheduler_interval: str | None = 'step', scheduler_monitor: str | None = 'validation/loss')

Configuration class for the optimizer and scheduler in the LightningModule.

__init__(optimizer_generator: Callable[..., optim.optimizer.Optimizer], scheduler_generator: Callable[..., optim.lr_scheduler.LRScheduler] | None = None, scheduler_interval: str | None = 'step', scheduler_monitor: str | None = 'validation/loss') → None

Methods

__init__(optimizer_generator[, ...])

Attributes

scheduler_generator
scheduler_interval
scheduler_monitor
optimizer_generator