tabensemb.model.AbstractNN.optimizers

method

AbstractNN.optimizers(use_pl_optimizer: bool = True) → Optimizer | LightningOptimizer | _FabricOptimizer | List[Optimizer] | List[LightningOptimizer] | List[_FabricOptimizer]

Returns the optimizer(s) that are being used during training. Useful for manual optimization.

Parameters:

use_pl_optimizer – If True, will wrap the optimizer(s) in a LightningOptimizer for automatic handling of precision and profiling.

Returns:

A single optimizer, or a list of optimizers in case multiple ones are present.