GitHub repository: mortfer/keras-gsam
Primary language: Python
Description: Surrogate Gap Guided Sharpness-Aware Minimization (GSAM) implementation for keras/tensorflow 2