├── .gitignore ├── CONTRIBUTING ├── LICENSE ├── OWNERS ├── README.md ├── __init__.py ├── autoaugment ├── __init__.py ├── autoaugment.py ├── autoaugment_test.py └── policies.py ├── figures ├── no_sam.png ├── sam_wide.png └── summary_plot.png ├── requirements.txt └── sam_jax ├── __init__.py ├── datasets ├── __init__.py ├── augmentation.py ├── dataset_source.py ├── dataset_source_imagenet.py ├── dataset_source_imagenet_test.py └── dataset_source_test.py ├── efficientnet ├── __init__.py ├── efficientnet.py ├── efficientnet_test.py ├── optim.py └── optim_test.py ├── imagenet_models ├── __init__.py ├── load_model.py ├── load_model_test.py ├── resnet.py └── resnet_test.py ├── models ├── __init__.py ├── load_model.py ├── load_model_test.py ├── pyramidnet.py ├── utils.py ├── wide_resnet.py └── wide_resnet_shakeshake.py ├── train.py └── training_utils ├── __init__.py ├── flax_training.py └── flax_training_test.py /.gitignore: -------------------------------------------------------------------------------- 1 | **/BUILD 2 | -------------------------------------------------------------------------------- /CONTRIBUTING: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/sam/HEAD/CONTRIBUTING -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/sam/HEAD/LICENSE -------------------------------------------------------------------------------- /OWNERS: -------------------------------------------------------------------------------- 1 | pierreforet 2 | akleiner 3 | -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/sam/HEAD/README.md -------------------------------------------------------------------------------- /__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/sam/HEAD/__init__.py -------------------------------------------------------------------------------- /autoaugment/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/sam/HEAD/autoaugment/__init__.py -------------------------------------------------------------------------------- /autoaugment/autoaugment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/sam/HEAD/autoaugment/autoaugment.py -------------------------------------------------------------------------------- /autoaugment/autoaugment_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/sam/HEAD/autoaugment/autoaugment_test.py -------------------------------------------------------------------------------- /autoaugment/policies.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/sam/HEAD/autoaugment/policies.py -------------------------------------------------------------------------------- /figures/no_sam.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/sam/HEAD/figures/no_sam.png -------------------------------------------------------------------------------- /figures/sam_wide.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/sam/HEAD/figures/sam_wide.png -------------------------------------------------------------------------------- /figures/summary_plot.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/sam/HEAD/figures/summary_plot.png -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/sam/HEAD/requirements.txt -------------------------------------------------------------------------------- /sam_jax/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/sam/HEAD/sam_jax/__init__.py -------------------------------------------------------------------------------- /sam_jax/datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/sam/HEAD/sam_jax/datasets/__init__.py -------------------------------------------------------------------------------- /sam_jax/datasets/augmentation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/sam/HEAD/sam_jax/datasets/augmentation.py -------------------------------------------------------------------------------- /sam_jax/datasets/dataset_source.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/sam/HEAD/sam_jax/datasets/dataset_source.py -------------------------------------------------------------------------------- /sam_jax/datasets/dataset_source_imagenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/sam/HEAD/sam_jax/datasets/dataset_source_imagenet.py -------------------------------------------------------------------------------- /sam_jax/datasets/dataset_source_imagenet_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/sam/HEAD/sam_jax/datasets/dataset_source_imagenet_test.py -------------------------------------------------------------------------------- /sam_jax/datasets/dataset_source_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/sam/HEAD/sam_jax/datasets/dataset_source_test.py -------------------------------------------------------------------------------- /sam_jax/efficientnet/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/sam/HEAD/sam_jax/efficientnet/__init__.py -------------------------------------------------------------------------------- /sam_jax/efficientnet/efficientnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/sam/HEAD/sam_jax/efficientnet/efficientnet.py -------------------------------------------------------------------------------- /sam_jax/efficientnet/efficientnet_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/sam/HEAD/sam_jax/efficientnet/efficientnet_test.py -------------------------------------------------------------------------------- /sam_jax/efficientnet/optim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/sam/HEAD/sam_jax/efficientnet/optim.py -------------------------------------------------------------------------------- /sam_jax/efficientnet/optim_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/sam/HEAD/sam_jax/efficientnet/optim_test.py -------------------------------------------------------------------------------- /sam_jax/imagenet_models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/sam/HEAD/sam_jax/imagenet_models/__init__.py -------------------------------------------------------------------------------- /sam_jax/imagenet_models/load_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/sam/HEAD/sam_jax/imagenet_models/load_model.py -------------------------------------------------------------------------------- /sam_jax/imagenet_models/load_model_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/sam/HEAD/sam_jax/imagenet_models/load_model_test.py -------------------------------------------------------------------------------- /sam_jax/imagenet_models/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/sam/HEAD/sam_jax/imagenet_models/resnet.py -------------------------------------------------------------------------------- /sam_jax/imagenet_models/resnet_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/sam/HEAD/sam_jax/imagenet_models/resnet_test.py -------------------------------------------------------------------------------- /sam_jax/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/sam/HEAD/sam_jax/models/__init__.py -------------------------------------------------------------------------------- /sam_jax/models/load_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/sam/HEAD/sam_jax/models/load_model.py -------------------------------------------------------------------------------- /sam_jax/models/load_model_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/sam/HEAD/sam_jax/models/load_model_test.py -------------------------------------------------------------------------------- /sam_jax/models/pyramidnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/sam/HEAD/sam_jax/models/pyramidnet.py -------------------------------------------------------------------------------- /sam_jax/models/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/sam/HEAD/sam_jax/models/utils.py -------------------------------------------------------------------------------- /sam_jax/models/wide_resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/sam/HEAD/sam_jax/models/wide_resnet.py -------------------------------------------------------------------------------- /sam_jax/models/wide_resnet_shakeshake.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/sam/HEAD/sam_jax/models/wide_resnet_shakeshake.py -------------------------------------------------------------------------------- /sam_jax/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/sam/HEAD/sam_jax/train.py -------------------------------------------------------------------------------- /sam_jax/training_utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/sam/HEAD/sam_jax/training_utils/__init__.py -------------------------------------------------------------------------------- /sam_jax/training_utils/flax_training.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/sam/HEAD/sam_jax/training_utils/flax_training.py -------------------------------------------------------------------------------- /sam_jax/training_utils/flax_training_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/sam/HEAD/sam_jax/training_utils/flax_training_test.py --------------------------------------------------------------------------------