/GSAM

PyTorch repository for ICLR 2022 paper (GSAM) which improves generalization (e.g. +3.8% top-1 accuracy on ImageNet with ViT-B/32)

Primary LanguagePythonMIT LicenseMIT

Issues