├── .gitattributes ├── .gitignore ├── .pre-commit-config.yaml ├── .vscode └── settings.json ├── LICENSE ├── README.md ├── figures └── result.jpg ├── lib ├── config │ └── default.yaml ├── core │ ├── evaluate.py │ ├── function.py │ └── loss.py ├── dataset │ ├── __init__.py │ ├── build.py │ ├── ffcv │ │ ├── make_dataloader.py │ │ └── write_datasets.py │ └── transforms │ │ ├── __init__.py │ │ ├── autoaugment.py │ │ ├── build.py │ │ ├── cutmix.py │ │ ├── cutout.py │ │ ├── mixup.py │ │ ├── randomcroppaste.py │ │ └── randomerase.py ├── models │ ├── __init__.py │ ├── att_overfit │ │ ├── __init__.py │ │ ├── att_net.py │ │ └── cbam.py │ ├── autorf │ │ ├── __init__.py │ │ ├── attention_structure.py │ │ ├── genotypes.py │ │ ├── models │ │ │ ├── __init__.py │ │ │ └── resnet20.py │ │ ├── operations.py │ │ └── spaces.py │ ├── build.py │ ├── cnn │ │ ├── ConvNeXt.py │ │ ├── __init__.py │ │ ├── attention.py │ │ ├── cbam_resnext.py │ │ ├── densenet.py │ │ ├── dla.py │ │ ├── dpn.py │ │ ├── efficientnetb0.py │ │ ├── genet.py │ │ ├── googlenet.py │ │ ├── inceptionv3.py │ │ ├── inceptionv4.py │ │ ├── lenet.py │ │ ├── mobilenet.py │ │ ├── mobilenetv2.py │ │ ├── nasnet.py │ │ ├── pnasnet.py │ │ ├── preact_resnet.py │ │ ├── regnet.py │ │ ├── resnet.py │ │ ├── resnet20.py │ │ ├── resnext.py │ │ ├── rir.py │ │ ├── senet.py │ │ ├── shake_shake.py │ │ ├── shufflenet.py │ │ ├── shufflenetv2.py │ │ ├── sknet.py │ │ ├── squeezenet.py │ │ ├── stochasticdepth.py │ │ ├── vgg.py │ │ ├── wideresidual.py │ │ └── xception.py │ ├── dawnnet.py │ ├── registry.py │ ├── spp_depth │ │ ├── __init__.py │ │ ├── poolings.py │ │ └── spp_resnet.py │ ├── toymodel.py │ └── vit │ │ ├── __init__.py │ │ ├── botnet.py │ │ ├── cct.py │ │ ├── ceit.py │ │ ├── coatnet.py │ │ ├── convnext.py │ │ ├── cotnet.py │ │ ├── cvt.py │ │ ├── mobile_vit.py │ │ ├── poolformer.py │ │ ├── swin_transformer.py │ │ ├── utils │ │ ├── embedder.py │ │ ├── helpers.py │ │ ├── layers.py │ │ ├── stochastic_depth.py │ │ ├── tokenizer.py │ │ ├── transformers.py │ │ └── utils.py │ │ ├── vision_transformer.py │ │ └── visual_attention_network.py ├── optim │ ├── __init__.py │ ├── adabound.py │ ├── adamw.py │ ├── asam.py │ ├── build.py │ └── lars.py ├── scheduler │ ├── __init__.py │ ├── build.py │ └── schdulers.py └── utils │ ├── Smodules.py │ ├── args.py │ ├── data_prefetch.py │ ├── ema.py │ ├── ghostbn.py │ ├── misc.py │ ├── rank.py │ ├── swa.py │ ├── utils.py │ ├── visualize.py │ └── visualize_RFs.py ├── requirements.txt ├── scripts ├── clean.py ├── retrain_spp.sh ├── run.bat ├── run.sh ├── run_att.sh ├── run_reject.bat ├── run_reject.sh └── run_spp.sh └── tools ├── _init_paths.py ├── lr_finder.py ├── reject.py ├── retrain.py ├── test.py └── train.py /.gitattributes: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pprp/PyTorch-CIFAR-Model-Hub/HEAD/.gitattributes -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pprp/PyTorch-CIFAR-Model-Hub/HEAD/.gitignore -------------------------------------------------------------------------------- /.pre-commit-config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pprp/PyTorch-CIFAR-Model-Hub/HEAD/.pre-commit-config.yaml -------------------------------------------------------------------------------- /.vscode/settings.json: -------------------------------------------------------------------------------- 1 | { 2 | "python.formatting.provider": "black" 3 | } 4 | -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pprp/PyTorch-CIFAR-Model-Hub/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pprp/PyTorch-CIFAR-Model-Hub/HEAD/README.md -------------------------------------------------------------------------------- /figures/result.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pprp/PyTorch-CIFAR-Model-Hub/HEAD/figures/result.jpg -------------------------------------------------------------------------------- /lib/config/default.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pprp/PyTorch-CIFAR-Model-Hub/HEAD/lib/config/default.yaml -------------------------------------------------------------------------------- /lib/core/evaluate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pprp/PyTorch-CIFAR-Model-Hub/HEAD/lib/core/evaluate.py -------------------------------------------------------------------------------- /lib/core/function.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pprp/PyTorch-CIFAR-Model-Hub/HEAD/lib/core/function.py -------------------------------------------------------------------------------- /lib/core/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pprp/PyTorch-CIFAR-Model-Hub/HEAD/lib/core/loss.py -------------------------------------------------------------------------------- /lib/dataset/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pprp/PyTorch-CIFAR-Model-Hub/HEAD/lib/dataset/__init__.py -------------------------------------------------------------------------------- /lib/dataset/build.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pprp/PyTorch-CIFAR-Model-Hub/HEAD/lib/dataset/build.py -------------------------------------------------------------------------------- /lib/dataset/ffcv/make_dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pprp/PyTorch-CIFAR-Model-Hub/HEAD/lib/dataset/ffcv/make_dataloader.py -------------------------------------------------------------------------------- /lib/dataset/ffcv/write_datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pprp/PyTorch-CIFAR-Model-Hub/HEAD/lib/dataset/ffcv/write_datasets.py -------------------------------------------------------------------------------- /lib/dataset/transforms/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pprp/PyTorch-CIFAR-Model-Hub/HEAD/lib/dataset/transforms/__init__.py -------------------------------------------------------------------------------- /lib/dataset/transforms/autoaugment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pprp/PyTorch-CIFAR-Model-Hub/HEAD/lib/dataset/transforms/autoaugment.py -------------------------------------------------------------------------------- /lib/dataset/transforms/build.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pprp/PyTorch-CIFAR-Model-Hub/HEAD/lib/dataset/transforms/build.py -------------------------------------------------------------------------------- /lib/dataset/transforms/cutmix.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pprp/PyTorch-CIFAR-Model-Hub/HEAD/lib/dataset/transforms/cutmix.py -------------------------------------------------------------------------------- /lib/dataset/transforms/cutout.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pprp/PyTorch-CIFAR-Model-Hub/HEAD/lib/dataset/transforms/cutout.py -------------------------------------------------------------------------------- /lib/dataset/transforms/mixup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pprp/PyTorch-CIFAR-Model-Hub/HEAD/lib/dataset/transforms/mixup.py -------------------------------------------------------------------------------- /lib/dataset/transforms/randomcroppaste.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pprp/PyTorch-CIFAR-Model-Hub/HEAD/lib/dataset/transforms/randomcroppaste.py -------------------------------------------------------------------------------- /lib/dataset/transforms/randomerase.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pprp/PyTorch-CIFAR-Model-Hub/HEAD/lib/dataset/transforms/randomerase.py -------------------------------------------------------------------------------- /lib/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pprp/PyTorch-CIFAR-Model-Hub/HEAD/lib/models/__init__.py -------------------------------------------------------------------------------- /lib/models/att_overfit/__init__.py: -------------------------------------------------------------------------------- 1 | from .att_net import * 2 | -------------------------------------------------------------------------------- /lib/models/att_overfit/att_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pprp/PyTorch-CIFAR-Model-Hub/HEAD/lib/models/att_overfit/att_net.py -------------------------------------------------------------------------------- /lib/models/att_overfit/cbam.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pprp/PyTorch-CIFAR-Model-Hub/HEAD/lib/models/att_overfit/cbam.py -------------------------------------------------------------------------------- /lib/models/autorf/__init__.py: -------------------------------------------------------------------------------- 1 | from .models import build_auto_network 2 | -------------------------------------------------------------------------------- /lib/models/autorf/attention_structure.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pprp/PyTorch-CIFAR-Model-Hub/HEAD/lib/models/autorf/attention_structure.py -------------------------------------------------------------------------------- /lib/models/autorf/genotypes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pprp/PyTorch-CIFAR-Model-Hub/HEAD/lib/models/autorf/genotypes.py -------------------------------------------------------------------------------- /lib/models/autorf/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pprp/PyTorch-CIFAR-Model-Hub/HEAD/lib/models/autorf/models/__init__.py -------------------------------------------------------------------------------- /lib/models/autorf/models/resnet20.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pprp/PyTorch-CIFAR-Model-Hub/HEAD/lib/models/autorf/models/resnet20.py -------------------------------------------------------------------------------- /lib/models/autorf/operations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pprp/PyTorch-CIFAR-Model-Hub/HEAD/lib/models/autorf/operations.py -------------------------------------------------------------------------------- /lib/models/autorf/spaces.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pprp/PyTorch-CIFAR-Model-Hub/HEAD/lib/models/autorf/spaces.py -------------------------------------------------------------------------------- /lib/models/build.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pprp/PyTorch-CIFAR-Model-Hub/HEAD/lib/models/build.py -------------------------------------------------------------------------------- /lib/models/cnn/ConvNeXt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pprp/PyTorch-CIFAR-Model-Hub/HEAD/lib/models/cnn/ConvNeXt.py -------------------------------------------------------------------------------- /lib/models/cnn/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pprp/PyTorch-CIFAR-Model-Hub/HEAD/lib/models/cnn/__init__.py -------------------------------------------------------------------------------- /lib/models/cnn/attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pprp/PyTorch-CIFAR-Model-Hub/HEAD/lib/models/cnn/attention.py -------------------------------------------------------------------------------- /lib/models/cnn/cbam_resnext.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pprp/PyTorch-CIFAR-Model-Hub/HEAD/lib/models/cnn/cbam_resnext.py -------------------------------------------------------------------------------- /lib/models/cnn/densenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pprp/PyTorch-CIFAR-Model-Hub/HEAD/lib/models/cnn/densenet.py -------------------------------------------------------------------------------- /lib/models/cnn/dla.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pprp/PyTorch-CIFAR-Model-Hub/HEAD/lib/models/cnn/dla.py -------------------------------------------------------------------------------- /lib/models/cnn/dpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pprp/PyTorch-CIFAR-Model-Hub/HEAD/lib/models/cnn/dpn.py -------------------------------------------------------------------------------- /lib/models/cnn/efficientnetb0.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pprp/PyTorch-CIFAR-Model-Hub/HEAD/lib/models/cnn/efficientnetb0.py -------------------------------------------------------------------------------- /lib/models/cnn/genet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pprp/PyTorch-CIFAR-Model-Hub/HEAD/lib/models/cnn/genet.py -------------------------------------------------------------------------------- /lib/models/cnn/googlenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pprp/PyTorch-CIFAR-Model-Hub/HEAD/lib/models/cnn/googlenet.py -------------------------------------------------------------------------------- /lib/models/cnn/inceptionv3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pprp/PyTorch-CIFAR-Model-Hub/HEAD/lib/models/cnn/inceptionv3.py -------------------------------------------------------------------------------- /lib/models/cnn/inceptionv4.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pprp/PyTorch-CIFAR-Model-Hub/HEAD/lib/models/cnn/inceptionv4.py -------------------------------------------------------------------------------- /lib/models/cnn/lenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pprp/PyTorch-CIFAR-Model-Hub/HEAD/lib/models/cnn/lenet.py -------------------------------------------------------------------------------- /lib/models/cnn/mobilenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pprp/PyTorch-CIFAR-Model-Hub/HEAD/lib/models/cnn/mobilenet.py -------------------------------------------------------------------------------- /lib/models/cnn/mobilenetv2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pprp/PyTorch-CIFAR-Model-Hub/HEAD/lib/models/cnn/mobilenetv2.py -------------------------------------------------------------------------------- /lib/models/cnn/nasnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pprp/PyTorch-CIFAR-Model-Hub/HEAD/lib/models/cnn/nasnet.py -------------------------------------------------------------------------------- /lib/models/cnn/pnasnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pprp/PyTorch-CIFAR-Model-Hub/HEAD/lib/models/cnn/pnasnet.py -------------------------------------------------------------------------------- /lib/models/cnn/preact_resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pprp/PyTorch-CIFAR-Model-Hub/HEAD/lib/models/cnn/preact_resnet.py -------------------------------------------------------------------------------- /lib/models/cnn/regnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pprp/PyTorch-CIFAR-Model-Hub/HEAD/lib/models/cnn/regnet.py -------------------------------------------------------------------------------- /lib/models/cnn/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pprp/PyTorch-CIFAR-Model-Hub/HEAD/lib/models/cnn/resnet.py -------------------------------------------------------------------------------- /lib/models/cnn/resnet20.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pprp/PyTorch-CIFAR-Model-Hub/HEAD/lib/models/cnn/resnet20.py -------------------------------------------------------------------------------- /lib/models/cnn/resnext.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pprp/PyTorch-CIFAR-Model-Hub/HEAD/lib/models/cnn/resnext.py -------------------------------------------------------------------------------- /lib/models/cnn/rir.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pprp/PyTorch-CIFAR-Model-Hub/HEAD/lib/models/cnn/rir.py -------------------------------------------------------------------------------- /lib/models/cnn/senet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pprp/PyTorch-CIFAR-Model-Hub/HEAD/lib/models/cnn/senet.py -------------------------------------------------------------------------------- /lib/models/cnn/shake_shake.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pprp/PyTorch-CIFAR-Model-Hub/HEAD/lib/models/cnn/shake_shake.py -------------------------------------------------------------------------------- /lib/models/cnn/shufflenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pprp/PyTorch-CIFAR-Model-Hub/HEAD/lib/models/cnn/shufflenet.py -------------------------------------------------------------------------------- /lib/models/cnn/shufflenetv2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pprp/PyTorch-CIFAR-Model-Hub/HEAD/lib/models/cnn/shufflenetv2.py -------------------------------------------------------------------------------- /lib/models/cnn/sknet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pprp/PyTorch-CIFAR-Model-Hub/HEAD/lib/models/cnn/sknet.py -------------------------------------------------------------------------------- /lib/models/cnn/squeezenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pprp/PyTorch-CIFAR-Model-Hub/HEAD/lib/models/cnn/squeezenet.py -------------------------------------------------------------------------------- /lib/models/cnn/stochasticdepth.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pprp/PyTorch-CIFAR-Model-Hub/HEAD/lib/models/cnn/stochasticdepth.py -------------------------------------------------------------------------------- /lib/models/cnn/vgg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pprp/PyTorch-CIFAR-Model-Hub/HEAD/lib/models/cnn/vgg.py -------------------------------------------------------------------------------- /lib/models/cnn/wideresidual.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pprp/PyTorch-CIFAR-Model-Hub/HEAD/lib/models/cnn/wideresidual.py -------------------------------------------------------------------------------- /lib/models/cnn/xception.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pprp/PyTorch-CIFAR-Model-Hub/HEAD/lib/models/cnn/xception.py -------------------------------------------------------------------------------- /lib/models/dawnnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pprp/PyTorch-CIFAR-Model-Hub/HEAD/lib/models/dawnnet.py -------------------------------------------------------------------------------- /lib/models/registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pprp/PyTorch-CIFAR-Model-Hub/HEAD/lib/models/registry.py -------------------------------------------------------------------------------- /lib/models/spp_depth/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pprp/PyTorch-CIFAR-Model-Hub/HEAD/lib/models/spp_depth/__init__.py -------------------------------------------------------------------------------- /lib/models/spp_depth/poolings.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pprp/PyTorch-CIFAR-Model-Hub/HEAD/lib/models/spp_depth/poolings.py -------------------------------------------------------------------------------- /lib/models/spp_depth/spp_resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pprp/PyTorch-CIFAR-Model-Hub/HEAD/lib/models/spp_depth/spp_resnet.py -------------------------------------------------------------------------------- /lib/models/toymodel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pprp/PyTorch-CIFAR-Model-Hub/HEAD/lib/models/toymodel.py -------------------------------------------------------------------------------- /lib/models/vit/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pprp/PyTorch-CIFAR-Model-Hub/HEAD/lib/models/vit/__init__.py -------------------------------------------------------------------------------- /lib/models/vit/botnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pprp/PyTorch-CIFAR-Model-Hub/HEAD/lib/models/vit/botnet.py -------------------------------------------------------------------------------- /lib/models/vit/cct.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pprp/PyTorch-CIFAR-Model-Hub/HEAD/lib/models/vit/cct.py -------------------------------------------------------------------------------- /lib/models/vit/ceit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pprp/PyTorch-CIFAR-Model-Hub/HEAD/lib/models/vit/ceit.py -------------------------------------------------------------------------------- /lib/models/vit/coatnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pprp/PyTorch-CIFAR-Model-Hub/HEAD/lib/models/vit/coatnet.py -------------------------------------------------------------------------------- /lib/models/vit/convnext.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pprp/PyTorch-CIFAR-Model-Hub/HEAD/lib/models/vit/convnext.py -------------------------------------------------------------------------------- /lib/models/vit/cotnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pprp/PyTorch-CIFAR-Model-Hub/HEAD/lib/models/vit/cotnet.py -------------------------------------------------------------------------------- /lib/models/vit/cvt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pprp/PyTorch-CIFAR-Model-Hub/HEAD/lib/models/vit/cvt.py -------------------------------------------------------------------------------- /lib/models/vit/mobile_vit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pprp/PyTorch-CIFAR-Model-Hub/HEAD/lib/models/vit/mobile_vit.py -------------------------------------------------------------------------------- /lib/models/vit/poolformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pprp/PyTorch-CIFAR-Model-Hub/HEAD/lib/models/vit/poolformer.py -------------------------------------------------------------------------------- /lib/models/vit/swin_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pprp/PyTorch-CIFAR-Model-Hub/HEAD/lib/models/vit/swin_transformer.py -------------------------------------------------------------------------------- /lib/models/vit/utils/embedder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pprp/PyTorch-CIFAR-Model-Hub/HEAD/lib/models/vit/utils/embedder.py -------------------------------------------------------------------------------- /lib/models/vit/utils/helpers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pprp/PyTorch-CIFAR-Model-Hub/HEAD/lib/models/vit/utils/helpers.py -------------------------------------------------------------------------------- /lib/models/vit/utils/layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pprp/PyTorch-CIFAR-Model-Hub/HEAD/lib/models/vit/utils/layers.py -------------------------------------------------------------------------------- /lib/models/vit/utils/stochastic_depth.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pprp/PyTorch-CIFAR-Model-Hub/HEAD/lib/models/vit/utils/stochastic_depth.py -------------------------------------------------------------------------------- /lib/models/vit/utils/tokenizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pprp/PyTorch-CIFAR-Model-Hub/HEAD/lib/models/vit/utils/tokenizer.py -------------------------------------------------------------------------------- /lib/models/vit/utils/transformers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pprp/PyTorch-CIFAR-Model-Hub/HEAD/lib/models/vit/utils/transformers.py -------------------------------------------------------------------------------- /lib/models/vit/utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pprp/PyTorch-CIFAR-Model-Hub/HEAD/lib/models/vit/utils/utils.py -------------------------------------------------------------------------------- /lib/models/vit/vision_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pprp/PyTorch-CIFAR-Model-Hub/HEAD/lib/models/vit/vision_transformer.py -------------------------------------------------------------------------------- /lib/models/vit/visual_attention_network.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pprp/PyTorch-CIFAR-Model-Hub/HEAD/lib/models/vit/visual_attention_network.py -------------------------------------------------------------------------------- /lib/optim/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pprp/PyTorch-CIFAR-Model-Hub/HEAD/lib/optim/__init__.py -------------------------------------------------------------------------------- /lib/optim/adabound.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pprp/PyTorch-CIFAR-Model-Hub/HEAD/lib/optim/adabound.py -------------------------------------------------------------------------------- /lib/optim/adamw.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pprp/PyTorch-CIFAR-Model-Hub/HEAD/lib/optim/adamw.py -------------------------------------------------------------------------------- /lib/optim/asam.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pprp/PyTorch-CIFAR-Model-Hub/HEAD/lib/optim/asam.py -------------------------------------------------------------------------------- /lib/optim/build.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pprp/PyTorch-CIFAR-Model-Hub/HEAD/lib/optim/build.py -------------------------------------------------------------------------------- /lib/optim/lars.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pprp/PyTorch-CIFAR-Model-Hub/HEAD/lib/optim/lars.py -------------------------------------------------------------------------------- /lib/scheduler/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pprp/PyTorch-CIFAR-Model-Hub/HEAD/lib/scheduler/__init__.py -------------------------------------------------------------------------------- /lib/scheduler/build.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pprp/PyTorch-CIFAR-Model-Hub/HEAD/lib/scheduler/build.py -------------------------------------------------------------------------------- /lib/scheduler/schdulers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pprp/PyTorch-CIFAR-Model-Hub/HEAD/lib/scheduler/schdulers.py -------------------------------------------------------------------------------- /lib/utils/Smodules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pprp/PyTorch-CIFAR-Model-Hub/HEAD/lib/utils/Smodules.py -------------------------------------------------------------------------------- /lib/utils/args.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pprp/PyTorch-CIFAR-Model-Hub/HEAD/lib/utils/args.py -------------------------------------------------------------------------------- /lib/utils/data_prefetch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pprp/PyTorch-CIFAR-Model-Hub/HEAD/lib/utils/data_prefetch.py -------------------------------------------------------------------------------- /lib/utils/ema.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pprp/PyTorch-CIFAR-Model-Hub/HEAD/lib/utils/ema.py -------------------------------------------------------------------------------- /lib/utils/ghostbn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pprp/PyTorch-CIFAR-Model-Hub/HEAD/lib/utils/ghostbn.py -------------------------------------------------------------------------------- /lib/utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pprp/PyTorch-CIFAR-Model-Hub/HEAD/lib/utils/misc.py -------------------------------------------------------------------------------- /lib/utils/rank.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pprp/PyTorch-CIFAR-Model-Hub/HEAD/lib/utils/rank.py -------------------------------------------------------------------------------- /lib/utils/swa.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pprp/PyTorch-CIFAR-Model-Hub/HEAD/lib/utils/swa.py -------------------------------------------------------------------------------- /lib/utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pprp/PyTorch-CIFAR-Model-Hub/HEAD/lib/utils/utils.py -------------------------------------------------------------------------------- /lib/utils/visualize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pprp/PyTorch-CIFAR-Model-Hub/HEAD/lib/utils/visualize.py -------------------------------------------------------------------------------- /lib/utils/visualize_RFs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pprp/PyTorch-CIFAR-Model-Hub/HEAD/lib/utils/visualize_RFs.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pprp/PyTorch-CIFAR-Model-Hub/HEAD/requirements.txt -------------------------------------------------------------------------------- /scripts/clean.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pprp/PyTorch-CIFAR-Model-Hub/HEAD/scripts/clean.py -------------------------------------------------------------------------------- /scripts/retrain_spp.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pprp/PyTorch-CIFAR-Model-Hub/HEAD/scripts/retrain_spp.sh -------------------------------------------------------------------------------- /scripts/run.bat: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pprp/PyTorch-CIFAR-Model-Hub/HEAD/scripts/run.bat -------------------------------------------------------------------------------- /scripts/run.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pprp/PyTorch-CIFAR-Model-Hub/HEAD/scripts/run.sh -------------------------------------------------------------------------------- /scripts/run_att.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pprp/PyTorch-CIFAR-Model-Hub/HEAD/scripts/run_att.sh -------------------------------------------------------------------------------- /scripts/run_reject.bat: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pprp/PyTorch-CIFAR-Model-Hub/HEAD/scripts/run_reject.bat -------------------------------------------------------------------------------- /scripts/run_reject.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pprp/PyTorch-CIFAR-Model-Hub/HEAD/scripts/run_reject.sh -------------------------------------------------------------------------------- /scripts/run_spp.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pprp/PyTorch-CIFAR-Model-Hub/HEAD/scripts/run_spp.sh -------------------------------------------------------------------------------- /tools/_init_paths.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pprp/PyTorch-CIFAR-Model-Hub/HEAD/tools/_init_paths.py -------------------------------------------------------------------------------- /tools/lr_finder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pprp/PyTorch-CIFAR-Model-Hub/HEAD/tools/lr_finder.py -------------------------------------------------------------------------------- /tools/reject.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pprp/PyTorch-CIFAR-Model-Hub/HEAD/tools/reject.py -------------------------------------------------------------------------------- /tools/retrain.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pprp/PyTorch-CIFAR-Model-Hub/HEAD/tools/retrain.py -------------------------------------------------------------------------------- /tools/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pprp/PyTorch-CIFAR-Model-Hub/HEAD/tools/test.py -------------------------------------------------------------------------------- /tools/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pprp/PyTorch-CIFAR-Model-Hub/HEAD/tools/train.py --------------------------------------------------------------------------------