GitHub repository: juntang-zhuang/GSAM
Primary language: Python
Description: PyTorch repository for ICLR 2022 paper (GSAM) which improves generalization (e.g. +3.8% top-1 accuracy on ImageNet with ViT-B/32)