pytorch-optimizer
pytorch-optimizer copied to clipboard
[Feature Request] SAM/ASAM Sharpness-Aware Minimization for Efficiently Improving Generalization
pytorch implementation https://github.com/davda54/sam
https://arxiv.org/abs/2010.01412 https://arxiv.org/abs/2102.11600