├── .flake8 ├── .gitignore ├── .pre-commit-config.yaml ├── LICENSE ├── README.md ├── config ├── analysis │ ├── data_augs_impact.yaml │ ├── final.yaml │ ├── full.yaml │ ├── limited.yaml │ └── none.yaml ├── cfg.yaml ├── data │ ├── CIFAR10-C.yaml │ ├── CIFAR10.yaml │ ├── CIFAR100.yaml │ ├── CINIC10.yaml │ ├── EMNIST.yaml │ ├── ImageNet.yaml │ ├── MNIST.yaml │ ├── SVHN.yaml │ ├── TinyImageNet.yaml │ └── db │ │ ├── LMDB.yaml │ │ └── none.yaml ├── hydra │ └── job_logging │ │ └── custom.yaml ├── hyp │ ├── _default_hyperparams.yaml │ ├── base_da.yaml │ ├── base_sgd.yaml │ ├── imagenet_A3.yaml │ ├── optim │ │ ├── adam.yaml │ │ ├── fista.yaml │ │ ├── gd.yaml │ │ ├── gd_agc.yaml │ │ ├── gd_clip.yaml │ │ └── lbfgs.yaml │ └── optim_modification │ │ ├── LARC.yaml │ │ ├── LARS.yaml │ │ ├── SAM.yaml │ │ └── none.yaml ├── impl │ ├── setup │ │ ├── distributed.yaml │ │ └── single.yaml │ └── standard.yaml └── model │ ├── convmixer.yaml │ ├── convmixer_small.yaml │ ├── densenet121.yaml │ ├── densenet161.yaml │ ├── densenet169.yaml │ ├── densenet201.yaml │ ├── e2cnn.yaml │ ├── e2cnn18.yaml │ ├── efficientnet.yaml │ ├── flipinvariantresnet.yaml │ ├── lieconv.yaml │ ├── linear.yaml │ ├── mlp.yaml │ ├── mobilenet.yaml │ ├── nfn.yaml │ ├── orbitresnet.yaml │ ├── pyramidnet110.yaml │ ├── pyramidnet272.yaml │ ├── resnet101.yaml │ ├── resnet110.yaml │ ├── resnet152.yaml │ ├── resnet18.yaml │ ├── resnet20-10.yaml │ ├── resnet20.yaml │ ├── resnet32.yaml │ ├── resnet34.yaml │ ├── resnet50.yaml │ ├── resnet56.yaml │ ├── resnet8.yaml │ ├── small_vit.yaml │ ├── small_vit2.yaml │ ├── swin.yaml │ ├── swinv2.yaml │ ├── vgg11.yaml │ ├── vgg13.yaml │ ├── vgg16.yaml │ └── vgg19.yaml ├── dataaug ├── __init__.py ├── analysis │ ├── __init__.py │ ├── analysis.py │ ├── cosine_pairs.py │ ├── rollouts.py │ └── welford.py ├── data │ ├── __init__.py │ ├── auto_augment.py │ ├── cached_dataset.py │ ├── cutout.py │ ├── data_preparation.py │ ├── datasets.py │ └── lmdb_datasets.py ├── models │ ├── __init__.py │ ├── convmixer.py │ ├── densenets.py │ ├── e2cnnmodels.py │ ├── mobilenet.py │ ├── models.py │ ├── modules.py │ ├── nfnets.py │ ├── pyramidnets.py │ ├── resnets.py │ ├── swin_transformers.py │ ├── utils.py │ ├── vgg.py │ └── vit_small.py ├── training │ ├── __init__.py │ ├── additional_optimizers │ │ ├── __init__.py │ │ ├── adaptive_clipping.py │ │ ├── fista.py │ │ ├── lars.py │ │ ├── lbfgs.py │ │ ├── sam.py │ │ ├── scheduler.py │ │ ├── sgd_agc.py │ │ └── sgd_linesearch.py │ ├── optimizers.py │ ├── training.py │ └── utils.py └── utils.py ├── environment_minimal.yml ├── fig0_scaling_baselines.sh ├── fig1_scaling_repetitions.sh ├── fig2_all_augmentations.sh ├── fig3a_scaling_model_width.sh ├── fig3b_scaling_model_type.sh ├── fig4_scaling_invariant_archs.sh ├── pyproject.toml ├── setup.cfg └── train_sgd_variant.py /.flake8: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JonasGeiping/dataaugs/HEAD/.flake8 -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JonasGeiping/dataaugs/HEAD/.gitignore -------------------------------------------------------------------------------- /.pre-commit-config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JonasGeiping/dataaugs/HEAD/.pre-commit-config.yaml -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JonasGeiping/dataaugs/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JonasGeiping/dataaugs/HEAD/README.md -------------------------------------------------------------------------------- /config/analysis/data_augs_impact.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JonasGeiping/dataaugs/HEAD/config/analysis/data_augs_impact.yaml -------------------------------------------------------------------------------- /config/analysis/final.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JonasGeiping/dataaugs/HEAD/config/analysis/final.yaml -------------------------------------------------------------------------------- /config/analysis/full.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JonasGeiping/dataaugs/HEAD/config/analysis/full.yaml -------------------------------------------------------------------------------- /config/analysis/limited.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JonasGeiping/dataaugs/HEAD/config/analysis/limited.yaml -------------------------------------------------------------------------------- /config/analysis/none.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JonasGeiping/dataaugs/HEAD/config/analysis/none.yaml -------------------------------------------------------------------------------- /config/cfg.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JonasGeiping/dataaugs/HEAD/config/cfg.yaml -------------------------------------------------------------------------------- /config/data/CIFAR10-C.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JonasGeiping/dataaugs/HEAD/config/data/CIFAR10-C.yaml -------------------------------------------------------------------------------- /config/data/CIFAR10.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JonasGeiping/dataaugs/HEAD/config/data/CIFAR10.yaml -------------------------------------------------------------------------------- /config/data/CIFAR100.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JonasGeiping/dataaugs/HEAD/config/data/CIFAR100.yaml -------------------------------------------------------------------------------- /config/data/CINIC10.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JonasGeiping/dataaugs/HEAD/config/data/CINIC10.yaml -------------------------------------------------------------------------------- /config/data/EMNIST.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JonasGeiping/dataaugs/HEAD/config/data/EMNIST.yaml -------------------------------------------------------------------------------- /config/data/ImageNet.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JonasGeiping/dataaugs/HEAD/config/data/ImageNet.yaml -------------------------------------------------------------------------------- /config/data/MNIST.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JonasGeiping/dataaugs/HEAD/config/data/MNIST.yaml -------------------------------------------------------------------------------- /config/data/SVHN.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JonasGeiping/dataaugs/HEAD/config/data/SVHN.yaml -------------------------------------------------------------------------------- /config/data/TinyImageNet.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JonasGeiping/dataaugs/HEAD/config/data/TinyImageNet.yaml -------------------------------------------------------------------------------- /config/data/db/LMDB.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JonasGeiping/dataaugs/HEAD/config/data/db/LMDB.yaml -------------------------------------------------------------------------------- /config/data/db/none.yaml: -------------------------------------------------------------------------------- 1 | 2 | name: 3 | -------------------------------------------------------------------------------- /config/hydra/job_logging/custom.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JonasGeiping/dataaugs/HEAD/config/hydra/job_logging/custom.yaml -------------------------------------------------------------------------------- /config/hyp/_default_hyperparams.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JonasGeiping/dataaugs/HEAD/config/hyp/_default_hyperparams.yaml -------------------------------------------------------------------------------- /config/hyp/base_da.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JonasGeiping/dataaugs/HEAD/config/hyp/base_da.yaml -------------------------------------------------------------------------------- /config/hyp/base_sgd.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JonasGeiping/dataaugs/HEAD/config/hyp/base_sgd.yaml -------------------------------------------------------------------------------- /config/hyp/imagenet_A3.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JonasGeiping/dataaugs/HEAD/config/hyp/imagenet_A3.yaml -------------------------------------------------------------------------------- /config/hyp/optim/adam.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JonasGeiping/dataaugs/HEAD/config/hyp/optim/adam.yaml -------------------------------------------------------------------------------- /config/hyp/optim/fista.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JonasGeiping/dataaugs/HEAD/config/hyp/optim/fista.yaml -------------------------------------------------------------------------------- /config/hyp/optim/gd.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JonasGeiping/dataaugs/HEAD/config/hyp/optim/gd.yaml -------------------------------------------------------------------------------- /config/hyp/optim/gd_agc.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JonasGeiping/dataaugs/HEAD/config/hyp/optim/gd_agc.yaml -------------------------------------------------------------------------------- /config/hyp/optim/gd_clip.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JonasGeiping/dataaugs/HEAD/config/hyp/optim/gd_clip.yaml -------------------------------------------------------------------------------- /config/hyp/optim/lbfgs.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JonasGeiping/dataaugs/HEAD/config/hyp/optim/lbfgs.yaml -------------------------------------------------------------------------------- /config/hyp/optim_modification/LARC.yaml: -------------------------------------------------------------------------------- 1 | 2 | name: LARC 3 | trust_coefficient: 0.02 4 | # clip: False 5 | eps: 1e-8 -------------------------------------------------------------------------------- /config/hyp/optim_modification/LARS.yaml: -------------------------------------------------------------------------------- 1 | 2 | name: LARS 3 | trust_coefficient: 0.02 4 | # clip: False 5 | eps: 1e-8 -------------------------------------------------------------------------------- /config/hyp/optim_modification/SAM.yaml: -------------------------------------------------------------------------------- 1 | 2 | name: SAM 3 | rho: 0.05 -------------------------------------------------------------------------------- /config/hyp/optim_modification/none.yaml: -------------------------------------------------------------------------------- 1 | 2 | name: none -------------------------------------------------------------------------------- /config/impl/setup/distributed.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JonasGeiping/dataaugs/HEAD/config/impl/setup/distributed.yaml -------------------------------------------------------------------------------- /config/impl/setup/single.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JonasGeiping/dataaugs/HEAD/config/impl/setup/single.yaml -------------------------------------------------------------------------------- /config/impl/standard.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JonasGeiping/dataaugs/HEAD/config/impl/standard.yaml -------------------------------------------------------------------------------- /config/model/convmixer.yaml: -------------------------------------------------------------------------------- 1 | 2 | name: ConvMixer 3 | depth: 16 4 | dim: 256 5 | kernel_size: 7 6 | -------------------------------------------------------------------------------- /config/model/convmixer_small.yaml: -------------------------------------------------------------------------------- 1 | 2 | name: ConvMixer 3 | depth: 8 4 | dim: 128 5 | kernel_size: 7 6 | -------------------------------------------------------------------------------- /config/model/densenet121.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JonasGeiping/dataaugs/HEAD/config/model/densenet121.yaml -------------------------------------------------------------------------------- /config/model/densenet161.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JonasGeiping/dataaugs/HEAD/config/model/densenet161.yaml -------------------------------------------------------------------------------- /config/model/densenet169.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JonasGeiping/dataaugs/HEAD/config/model/densenet169.yaml -------------------------------------------------------------------------------- /config/model/densenet201.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JonasGeiping/dataaugs/HEAD/config/model/densenet201.yaml -------------------------------------------------------------------------------- /config/model/e2cnn.yaml: -------------------------------------------------------------------------------- 1 | 2 | name: e2cnn-base-v2 3 | depth: 22 4 | width_modifier: 4 5 | -------------------------------------------------------------------------------- /config/model/e2cnn18.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JonasGeiping/dataaugs/HEAD/config/model/e2cnn18.yaml -------------------------------------------------------------------------------- /config/model/efficientnet.yaml: -------------------------------------------------------------------------------- 1 | name: efficientnet 2 | -------------------------------------------------------------------------------- /config/model/flipinvariantresnet.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JonasGeiping/dataaugs/HEAD/config/model/flipinvariantresnet.yaml -------------------------------------------------------------------------------- /config/model/lieconv.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JonasGeiping/dataaugs/HEAD/config/model/lieconv.yaml -------------------------------------------------------------------------------- /config/model/linear.yaml: -------------------------------------------------------------------------------- 1 | 2 | name: linear 3 | # only use this for debugging -------------------------------------------------------------------------------- /config/model/mlp.yaml: -------------------------------------------------------------------------------- 1 | name: mlp 2 | width: 1024 3 | -------------------------------------------------------------------------------- /config/model/mobilenet.yaml: -------------------------------------------------------------------------------- 1 | name: MobileNetV2 2 | -------------------------------------------------------------------------------- /config/model/nfn.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JonasGeiping/dataaugs/HEAD/config/model/nfn.yaml -------------------------------------------------------------------------------- /config/model/orbitresnet.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JonasGeiping/dataaugs/HEAD/config/model/orbitresnet.yaml -------------------------------------------------------------------------------- /config/model/pyramidnet110.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JonasGeiping/dataaugs/HEAD/config/model/pyramidnet110.yaml -------------------------------------------------------------------------------- /config/model/pyramidnet272.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JonasGeiping/dataaugs/HEAD/config/model/pyramidnet272.yaml -------------------------------------------------------------------------------- /config/model/resnet101.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JonasGeiping/dataaugs/HEAD/config/model/resnet101.yaml -------------------------------------------------------------------------------- /config/model/resnet110.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JonasGeiping/dataaugs/HEAD/config/model/resnet110.yaml -------------------------------------------------------------------------------- /config/model/resnet152.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JonasGeiping/dataaugs/HEAD/config/model/resnet152.yaml -------------------------------------------------------------------------------- /config/model/resnet18.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JonasGeiping/dataaugs/HEAD/config/model/resnet18.yaml -------------------------------------------------------------------------------- /config/model/resnet20-10.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JonasGeiping/dataaugs/HEAD/config/model/resnet20-10.yaml -------------------------------------------------------------------------------- /config/model/resnet20.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JonasGeiping/dataaugs/HEAD/config/model/resnet20.yaml -------------------------------------------------------------------------------- /config/model/resnet32.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JonasGeiping/dataaugs/HEAD/config/model/resnet32.yaml -------------------------------------------------------------------------------- /config/model/resnet34.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JonasGeiping/dataaugs/HEAD/config/model/resnet34.yaml -------------------------------------------------------------------------------- /config/model/resnet50.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JonasGeiping/dataaugs/HEAD/config/model/resnet50.yaml -------------------------------------------------------------------------------- /config/model/resnet56.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JonasGeiping/dataaugs/HEAD/config/model/resnet56.yaml -------------------------------------------------------------------------------- /config/model/resnet8.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JonasGeiping/dataaugs/HEAD/config/model/resnet8.yaml -------------------------------------------------------------------------------- /config/model/small_vit.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JonasGeiping/dataaugs/HEAD/config/model/small_vit.yaml -------------------------------------------------------------------------------- /config/model/small_vit2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JonasGeiping/dataaugs/HEAD/config/model/small_vit2.yaml -------------------------------------------------------------------------------- /config/model/swin.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JonasGeiping/dataaugs/HEAD/config/model/swin.yaml -------------------------------------------------------------------------------- /config/model/swinv2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JonasGeiping/dataaugs/HEAD/config/model/swinv2.yaml -------------------------------------------------------------------------------- /config/model/vgg11.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JonasGeiping/dataaugs/HEAD/config/model/vgg11.yaml -------------------------------------------------------------------------------- /config/model/vgg13.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JonasGeiping/dataaugs/HEAD/config/model/vgg13.yaml -------------------------------------------------------------------------------- /config/model/vgg16.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JonasGeiping/dataaugs/HEAD/config/model/vgg16.yaml -------------------------------------------------------------------------------- /config/model/vgg19.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JonasGeiping/dataaugs/HEAD/config/model/vgg19.yaml -------------------------------------------------------------------------------- /dataaug/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JonasGeiping/dataaugs/HEAD/dataaug/__init__.py -------------------------------------------------------------------------------- /dataaug/analysis/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JonasGeiping/dataaugs/HEAD/dataaug/analysis/__init__.py -------------------------------------------------------------------------------- /dataaug/analysis/analysis.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JonasGeiping/dataaugs/HEAD/dataaug/analysis/analysis.py -------------------------------------------------------------------------------- /dataaug/analysis/cosine_pairs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JonasGeiping/dataaugs/HEAD/dataaug/analysis/cosine_pairs.py -------------------------------------------------------------------------------- /dataaug/analysis/rollouts.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JonasGeiping/dataaugs/HEAD/dataaug/analysis/rollouts.py -------------------------------------------------------------------------------- /dataaug/analysis/welford.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JonasGeiping/dataaugs/HEAD/dataaug/analysis/welford.py -------------------------------------------------------------------------------- /dataaug/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JonasGeiping/dataaugs/HEAD/dataaug/data/__init__.py -------------------------------------------------------------------------------- /dataaug/data/auto_augment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JonasGeiping/dataaugs/HEAD/dataaug/data/auto_augment.py -------------------------------------------------------------------------------- /dataaug/data/cached_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JonasGeiping/dataaugs/HEAD/dataaug/data/cached_dataset.py -------------------------------------------------------------------------------- /dataaug/data/cutout.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JonasGeiping/dataaugs/HEAD/dataaug/data/cutout.py -------------------------------------------------------------------------------- /dataaug/data/data_preparation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JonasGeiping/dataaugs/HEAD/dataaug/data/data_preparation.py -------------------------------------------------------------------------------- /dataaug/data/datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JonasGeiping/dataaugs/HEAD/dataaug/data/datasets.py -------------------------------------------------------------------------------- /dataaug/data/lmdb_datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JonasGeiping/dataaugs/HEAD/dataaug/data/lmdb_datasets.py -------------------------------------------------------------------------------- /dataaug/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JonasGeiping/dataaugs/HEAD/dataaug/models/__init__.py -------------------------------------------------------------------------------- /dataaug/models/convmixer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JonasGeiping/dataaugs/HEAD/dataaug/models/convmixer.py -------------------------------------------------------------------------------- /dataaug/models/densenets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JonasGeiping/dataaugs/HEAD/dataaug/models/densenets.py -------------------------------------------------------------------------------- /dataaug/models/e2cnnmodels.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JonasGeiping/dataaugs/HEAD/dataaug/models/e2cnnmodels.py -------------------------------------------------------------------------------- /dataaug/models/mobilenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JonasGeiping/dataaugs/HEAD/dataaug/models/mobilenet.py -------------------------------------------------------------------------------- /dataaug/models/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JonasGeiping/dataaugs/HEAD/dataaug/models/models.py -------------------------------------------------------------------------------- /dataaug/models/modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JonasGeiping/dataaugs/HEAD/dataaug/models/modules.py -------------------------------------------------------------------------------- /dataaug/models/nfnets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JonasGeiping/dataaugs/HEAD/dataaug/models/nfnets.py -------------------------------------------------------------------------------- /dataaug/models/pyramidnets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JonasGeiping/dataaugs/HEAD/dataaug/models/pyramidnets.py -------------------------------------------------------------------------------- /dataaug/models/resnets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JonasGeiping/dataaugs/HEAD/dataaug/models/resnets.py -------------------------------------------------------------------------------- /dataaug/models/swin_transformers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JonasGeiping/dataaugs/HEAD/dataaug/models/swin_transformers.py -------------------------------------------------------------------------------- /dataaug/models/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JonasGeiping/dataaugs/HEAD/dataaug/models/utils.py -------------------------------------------------------------------------------- /dataaug/models/vgg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JonasGeiping/dataaugs/HEAD/dataaug/models/vgg.py -------------------------------------------------------------------------------- /dataaug/models/vit_small.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JonasGeiping/dataaugs/HEAD/dataaug/models/vit_small.py -------------------------------------------------------------------------------- /dataaug/training/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JonasGeiping/dataaugs/HEAD/dataaug/training/__init__.py -------------------------------------------------------------------------------- /dataaug/training/additional_optimizers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JonasGeiping/dataaugs/HEAD/dataaug/training/additional_optimizers/__init__.py -------------------------------------------------------------------------------- /dataaug/training/additional_optimizers/adaptive_clipping.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JonasGeiping/dataaugs/HEAD/dataaug/training/additional_optimizers/adaptive_clipping.py -------------------------------------------------------------------------------- /dataaug/training/additional_optimizers/fista.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JonasGeiping/dataaugs/HEAD/dataaug/training/additional_optimizers/fista.py -------------------------------------------------------------------------------- /dataaug/training/additional_optimizers/lars.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JonasGeiping/dataaugs/HEAD/dataaug/training/additional_optimizers/lars.py -------------------------------------------------------------------------------- /dataaug/training/additional_optimizers/lbfgs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JonasGeiping/dataaugs/HEAD/dataaug/training/additional_optimizers/lbfgs.py -------------------------------------------------------------------------------- /dataaug/training/additional_optimizers/sam.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JonasGeiping/dataaugs/HEAD/dataaug/training/additional_optimizers/sam.py -------------------------------------------------------------------------------- /dataaug/training/additional_optimizers/scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JonasGeiping/dataaugs/HEAD/dataaug/training/additional_optimizers/scheduler.py -------------------------------------------------------------------------------- /dataaug/training/additional_optimizers/sgd_agc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JonasGeiping/dataaugs/HEAD/dataaug/training/additional_optimizers/sgd_agc.py -------------------------------------------------------------------------------- /dataaug/training/additional_optimizers/sgd_linesearch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JonasGeiping/dataaugs/HEAD/dataaug/training/additional_optimizers/sgd_linesearch.py -------------------------------------------------------------------------------- /dataaug/training/optimizers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JonasGeiping/dataaugs/HEAD/dataaug/training/optimizers.py -------------------------------------------------------------------------------- /dataaug/training/training.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JonasGeiping/dataaugs/HEAD/dataaug/training/training.py -------------------------------------------------------------------------------- /dataaug/training/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JonasGeiping/dataaugs/HEAD/dataaug/training/utils.py -------------------------------------------------------------------------------- /dataaug/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JonasGeiping/dataaugs/HEAD/dataaug/utils.py -------------------------------------------------------------------------------- /environment_minimal.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JonasGeiping/dataaugs/HEAD/environment_minimal.yml -------------------------------------------------------------------------------- /fig0_scaling_baselines.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JonasGeiping/dataaugs/HEAD/fig0_scaling_baselines.sh -------------------------------------------------------------------------------- /fig1_scaling_repetitions.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JonasGeiping/dataaugs/HEAD/fig1_scaling_repetitions.sh -------------------------------------------------------------------------------- /fig2_all_augmentations.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JonasGeiping/dataaugs/HEAD/fig2_all_augmentations.sh -------------------------------------------------------------------------------- /fig3a_scaling_model_width.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JonasGeiping/dataaugs/HEAD/fig3a_scaling_model_width.sh -------------------------------------------------------------------------------- /fig3b_scaling_model_type.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JonasGeiping/dataaugs/HEAD/fig3b_scaling_model_type.sh -------------------------------------------------------------------------------- /fig4_scaling_invariant_archs.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JonasGeiping/dataaugs/HEAD/fig4_scaling_invariant_archs.sh -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JonasGeiping/dataaugs/HEAD/pyproject.toml -------------------------------------------------------------------------------- /setup.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JonasGeiping/dataaugs/HEAD/setup.cfg -------------------------------------------------------------------------------- /train_sgd_variant.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JonasGeiping/dataaugs/HEAD/train_sgd_variant.py --------------------------------------------------------------------------------