← Back to API reference

neuraltrain.optimizers.base.LightningOptimizer¶

pydantic model neuraltrain.optimizers.base.LightningOptimizer[source][source]¶

Pydantic configuration for Lightning optimizer.

Fields:

interval (Literal['step', 'epoch'])
optimizer (neuraltrain.optimizers.base.BaseOptimizer)
scheduler (neuraltrain.optimizers.base.BaseLRScheduler | None)

field optimizer: BaseOptimizer [Required][source]¶

field scheduler: BaseLRScheduler | None = None[source]¶

field interval: Literal['step', 'epoch'] = 'step'[source]¶

build(params: Iterable[Tensor], **scheduler_build_kwargs: Any) → dict[str, Any][source][source]¶

← Back to API reference