Skip to content

Supported Optimizers

Source

Optimizers and schedulers supported by CAREamics.

SupportedOptimizer

Bases: StrEnum

Supported optimizers.

Attributes:

  • Adam (str) –

    Adam optimizer.

  • AdamW (str) –

    AdamW optimizer.

  • Adamax (str) –

    Adamax optimizer.

  • SGD (str) –

    Stochastic Gradient Descent optimizer.

SupportedScheduler

Bases: StrEnum

Supported schedulers.

Attributes:

  • ReduceLROnPlateau (str) –

    Reduce learning rate on plateau.

  • StepLR (str) –

    Step learning rate.