├── .gitignore ├── LICENSE ├── README.md ├── __init__.py ├── config ├── __init__.py ├── base_config.py └── train_config.py ├── ds_pipeline ├── __init__.py ├── autoaugment │ ├── __init__.py │ ├── autoaugment.py │ ├── autoaugment_test.py │ └── policies.py ├── datasets │ ├── __init__.py │ ├── augmentation.py │ ├── dataset_source.py │ └── dataset_source_imagenet.py └── get_dataset.py ├── main ├── __init__.py ├── main.py └── util.py ├── models ├── __init__.py ├── available_models │ ├── __init__.py │ ├── cait.py │ ├── pyramidnet.py │ ├── pyramidnet_shakedrop.py │ ├── resnet.py │ ├── swin_transformer.py │ ├── util.py │ ├── vgg.py │ ├── vit_model.py │ ├── wrn.py │ └── wrn_shakeshake.py └── load_model.py ├── optimizer ├── __init__.py └── get_optimizer.py ├── recipes ├── pyramidnet-sd-cifar.sh ├── resnet-cifar.sh ├── resnet-imagenet.sh ├── vggbn-imagenet.sh ├── vit-cifar-basic-grwarmup.sh ├── vit-cifar-basic.sh ├── vit_cifar_heavy.sh ├── wideresnet-cifar.sh └── wideresnet-ss-cifar.sh ├── requirements.txt └── training ├── __init__.py ├── lr_schedule.py ├── training.py ├── training_core.py └── utli.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaoyang-0204/gnp/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaoyang-0204/gnp/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaoyang-0204/gnp/HEAD/README.md -------------------------------------------------------------------------------- /__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaoyang-0204/gnp/HEAD/__init__.py -------------------------------------------------------------------------------- /config/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /config/base_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaoyang-0204/gnp/HEAD/config/base_config.py -------------------------------------------------------------------------------- /config/train_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaoyang-0204/gnp/HEAD/config/train_config.py -------------------------------------------------------------------------------- /ds_pipeline/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaoyang-0204/gnp/HEAD/ds_pipeline/__init__.py -------------------------------------------------------------------------------- /ds_pipeline/autoaugment/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaoyang-0204/gnp/HEAD/ds_pipeline/autoaugment/__init__.py -------------------------------------------------------------------------------- /ds_pipeline/autoaugment/autoaugment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaoyang-0204/gnp/HEAD/ds_pipeline/autoaugment/autoaugment.py -------------------------------------------------------------------------------- /ds_pipeline/autoaugment/autoaugment_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaoyang-0204/gnp/HEAD/ds_pipeline/autoaugment/autoaugment_test.py -------------------------------------------------------------------------------- /ds_pipeline/autoaugment/policies.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaoyang-0204/gnp/HEAD/ds_pipeline/autoaugment/policies.py -------------------------------------------------------------------------------- /ds_pipeline/datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaoyang-0204/gnp/HEAD/ds_pipeline/datasets/__init__.py -------------------------------------------------------------------------------- /ds_pipeline/datasets/augmentation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaoyang-0204/gnp/HEAD/ds_pipeline/datasets/augmentation.py -------------------------------------------------------------------------------- /ds_pipeline/datasets/dataset_source.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaoyang-0204/gnp/HEAD/ds_pipeline/datasets/dataset_source.py -------------------------------------------------------------------------------- /ds_pipeline/datasets/dataset_source_imagenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaoyang-0204/gnp/HEAD/ds_pipeline/datasets/dataset_source_imagenet.py -------------------------------------------------------------------------------- /ds_pipeline/get_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaoyang-0204/gnp/HEAD/ds_pipeline/get_dataset.py -------------------------------------------------------------------------------- /main/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaoyang-0204/gnp/HEAD/main/__init__.py -------------------------------------------------------------------------------- /main/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaoyang-0204/gnp/HEAD/main/main.py -------------------------------------------------------------------------------- /main/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaoyang-0204/gnp/HEAD/main/util.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaoyang-0204/gnp/HEAD/models/__init__.py -------------------------------------------------------------------------------- /models/available_models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaoyang-0204/gnp/HEAD/models/available_models/__init__.py -------------------------------------------------------------------------------- /models/available_models/cait.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaoyang-0204/gnp/HEAD/models/available_models/cait.py -------------------------------------------------------------------------------- /models/available_models/pyramidnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaoyang-0204/gnp/HEAD/models/available_models/pyramidnet.py -------------------------------------------------------------------------------- /models/available_models/pyramidnet_shakedrop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaoyang-0204/gnp/HEAD/models/available_models/pyramidnet_shakedrop.py -------------------------------------------------------------------------------- /models/available_models/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaoyang-0204/gnp/HEAD/models/available_models/resnet.py -------------------------------------------------------------------------------- /models/available_models/swin_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaoyang-0204/gnp/HEAD/models/available_models/swin_transformer.py -------------------------------------------------------------------------------- /models/available_models/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaoyang-0204/gnp/HEAD/models/available_models/util.py -------------------------------------------------------------------------------- /models/available_models/vgg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaoyang-0204/gnp/HEAD/models/available_models/vgg.py -------------------------------------------------------------------------------- /models/available_models/vit_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaoyang-0204/gnp/HEAD/models/available_models/vit_model.py -------------------------------------------------------------------------------- /models/available_models/wrn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaoyang-0204/gnp/HEAD/models/available_models/wrn.py -------------------------------------------------------------------------------- /models/available_models/wrn_shakeshake.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaoyang-0204/gnp/HEAD/models/available_models/wrn_shakeshake.py -------------------------------------------------------------------------------- /models/load_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaoyang-0204/gnp/HEAD/models/load_model.py -------------------------------------------------------------------------------- /optimizer/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaoyang-0204/gnp/HEAD/optimizer/__init__.py -------------------------------------------------------------------------------- /optimizer/get_optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaoyang-0204/gnp/HEAD/optimizer/get_optimizer.py -------------------------------------------------------------------------------- /recipes/pyramidnet-sd-cifar.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaoyang-0204/gnp/HEAD/recipes/pyramidnet-sd-cifar.sh -------------------------------------------------------------------------------- /recipes/resnet-cifar.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaoyang-0204/gnp/HEAD/recipes/resnet-cifar.sh -------------------------------------------------------------------------------- /recipes/resnet-imagenet.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaoyang-0204/gnp/HEAD/recipes/resnet-imagenet.sh -------------------------------------------------------------------------------- /recipes/vggbn-imagenet.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaoyang-0204/gnp/HEAD/recipes/vggbn-imagenet.sh -------------------------------------------------------------------------------- /recipes/vit-cifar-basic-grwarmup.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaoyang-0204/gnp/HEAD/recipes/vit-cifar-basic-grwarmup.sh -------------------------------------------------------------------------------- /recipes/vit-cifar-basic.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaoyang-0204/gnp/HEAD/recipes/vit-cifar-basic.sh -------------------------------------------------------------------------------- /recipes/vit_cifar_heavy.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaoyang-0204/gnp/HEAD/recipes/vit_cifar_heavy.sh -------------------------------------------------------------------------------- /recipes/wideresnet-cifar.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaoyang-0204/gnp/HEAD/recipes/wideresnet-cifar.sh -------------------------------------------------------------------------------- /recipes/wideresnet-ss-cifar.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaoyang-0204/gnp/HEAD/recipes/wideresnet-ss-cifar.sh -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaoyang-0204/gnp/HEAD/requirements.txt -------------------------------------------------------------------------------- /training/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaoyang-0204/gnp/HEAD/training/__init__.py -------------------------------------------------------------------------------- /training/lr_schedule.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaoyang-0204/gnp/HEAD/training/lr_schedule.py -------------------------------------------------------------------------------- /training/training.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaoyang-0204/gnp/HEAD/training/training.py -------------------------------------------------------------------------------- /training/training_core.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaoyang-0204/gnp/HEAD/training/training_core.py -------------------------------------------------------------------------------- /training/utli.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhaoyang-0204/gnp/HEAD/training/utli.py --------------------------------------------------------------------------------