├── .gitignore ├── README.md ├── data.py ├── eval_sharpness.py ├── images ├── abstract.png └── summary_slide.png ├── models.py ├── notebooks ├── sharpness_linear_model.ipynb └── sharpness_vs_generalization.ipynb ├── randomaug.py ├── requirements.txt ├── sh └── eval_sharpness_final.sh ├── sharpness.py └── utils.py /.gitignore: -------------------------------------------------------------------------------- 1 | __pycache__ 2 | models/ 3 | logs* 4 | data/ 5 | plots/ -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tml-epfl/sharpness-vs-generalization/HEAD/README.md -------------------------------------------------------------------------------- /data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tml-epfl/sharpness-vs-generalization/HEAD/data.py -------------------------------------------------------------------------------- /eval_sharpness.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tml-epfl/sharpness-vs-generalization/HEAD/eval_sharpness.py -------------------------------------------------------------------------------- /images/abstract.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tml-epfl/sharpness-vs-generalization/HEAD/images/abstract.png -------------------------------------------------------------------------------- /images/summary_slide.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tml-epfl/sharpness-vs-generalization/HEAD/images/summary_slide.png -------------------------------------------------------------------------------- /models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tml-epfl/sharpness-vs-generalization/HEAD/models.py -------------------------------------------------------------------------------- /notebooks/sharpness_linear_model.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tml-epfl/sharpness-vs-generalization/HEAD/notebooks/sharpness_linear_model.ipynb -------------------------------------------------------------------------------- /notebooks/sharpness_vs_generalization.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tml-epfl/sharpness-vs-generalization/HEAD/notebooks/sharpness_vs_generalization.ipynb -------------------------------------------------------------------------------- /randomaug.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tml-epfl/sharpness-vs-generalization/HEAD/randomaug.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tml-epfl/sharpness-vs-generalization/HEAD/requirements.txt -------------------------------------------------------------------------------- /sh/eval_sharpness_final.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tml-epfl/sharpness-vs-generalization/HEAD/sh/eval_sharpness_final.sh -------------------------------------------------------------------------------- /sharpness.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tml-epfl/sharpness-vs-generalization/HEAD/sharpness.py -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tml-epfl/sharpness-vs-generalization/HEAD/utils.py --------------------------------------------------------------------------------