/sam

SAM: Sharpness-Aware Minimization (PyTorch)

Primary LanguagePythonMIT LicenseMIT

Watchers