Skip to content

Supported Optimizers

Source

Optimizers and schedulers supported by CAREamics.

SupportedOptimizer

Bases: str, BaseEnum

Supported optimizers.

Attributes:

Name Type Description
Adam str

Adam optimizer.

AdamW str

AdamW optimizer.

Adamax str

Adamax optimizer.

SGD str

Stochastic Gradient Descent optimizer.

SupportedScheduler

Bases: str, BaseEnum

Supported schedulers.

Attributes:

Name Type Description
ReduceLROnPlateau str

Reduce learning rate on plateau.

StepLR str

Step learning rate.