composer.algorithms.sam#
Modules
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
Adds sharpness-aware minimization (Foret et al, 2020) by wrapping an existing optimizer with a |
|
Wraps an optimizer with sharpness-aware minimization (Foret et al, 2020). |