├── .gitignore ├── README.md ├── SAM.ipynb ├── resnet_cifar10.py └── utils.py /.gitignore: -------------------------------------------------------------------------------- 1 | .idea 2 | .ipynb_checkpoints/ 3 | .DS_Store 4 | -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sayakpaul/Sharpness-Aware-Minimization-TensorFlow/HEAD/README.md -------------------------------------------------------------------------------- /SAM.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sayakpaul/Sharpness-Aware-Minimization-TensorFlow/HEAD/SAM.ipynb -------------------------------------------------------------------------------- /resnet_cifar10.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sayakpaul/Sharpness-Aware-Minimization-TensorFlow/HEAD/resnet_cifar10.py -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sayakpaul/Sharpness-Aware-Minimization-TensorFlow/HEAD/utils.py --------------------------------------------------------------------------------