sam

Modules

composer.algorithms.sam.sam

SAM algorithm and optimizer class.

SAM (Foret et al, 2020) wraps an existing optimizer with a SAMOptimizer which makes the optimizer minimize both loss value and sharpness.This can improves model generalization and provide robustness to label noise.

See the Method Card for more details.

Classes

SAM	Adds sharpness-aware minimization (Foret et al, 2020) by wrapping an existing optimizer with a SAMOptimizer.
SAMOptimizer	Wraps an optimizer with sharpness-aware minimization (Foret et al, 2020).