├── .gitignore ├── LICENSE ├── README.md ├── bake.png ├── imagenet ├── LICENSE ├── README.md ├── configs │ ├── effnet │ │ ├── BAKE-EN-B0_dds_8gpu.yaml │ │ └── EN-B0_dds_8gpu.yaml │ ├── mobilenet │ │ ├── BAKE-M-V2-W1_dds_4gpu.yaml │ │ └── M-V2-W1_dds_4gpu.yaml │ ├── resnest │ │ ├── BAKE-S-101_dds_8gpu.yaml │ │ ├── BAKE-S-50_dds_8gpu.yaml │ │ ├── S-101_dds_8gpu.yaml │ │ └── S-50_dds_8gpu.yaml │ ├── resnet │ │ ├── BAKE-R-101-1x64d_dds_8gpu.yaml │ │ ├── BAKE-R-152-1x64d_dds_8gpu.yaml │ │ ├── BAKE-R-50-1x64d_dds_8gpu.yaml │ │ ├── R-101-1x64d_dds_8gpu.yaml │ │ ├── R-152-1x64d_dds_8gpu.yaml │ │ └── R-50-1x64d_dds_8gpu.yaml │ └── resnext │ │ ├── BAKE-X-101-32x4d_dds_8gpu.yaml │ │ ├── BAKE-X-152-32x4d_dds_8gpu.yaml │ │ ├── X-101-32x4d_dds_8gpu.yaml │ │ └── X-152-32x4d_dds_8gpu.yaml ├── pycls │ ├── __init__.py │ ├── core │ │ ├── __init__.py │ │ ├── benchmark.py │ │ ├── builders.py │ │ ├── checkpoint.py │ │ ├── config.py │ │ ├── distributed.py │ │ ├── io.py │ │ ├── logging.py │ │ ├── meters.py │ │ ├── net.py │ │ ├── optimizer.py │ │ ├── plotting.py │ │ ├── timer.py │ │ └── trainer.py │ ├── datasets │ │ ├── __init__.py │ │ ├── imagenet.py │ │ ├── loader.py │ │ ├── sampler.py │ │ └── transforms.py │ └── models │ │ ├── __init__.py │ │ ├── anynet.py │ │ ├── effnet.py │ │ ├── mobilenetv2.py │ │ ├── resnest.py │ │ └── resnet.py ├── requirements.txt ├── setup.py └── tools │ ├── dist_test.sh │ ├── dist_train.sh │ ├── slurm_test.sh │ ├── slurm_train.sh │ ├── test_net.py │ └── train_net.py └── small_scale ├── README.md ├── data ├── README.md └── tools │ ├── cub.py │ ├── mit67.py │ ├── stanford_dogs.py │ └── tinyimagenet.py ├── datasets.py ├── models ├── __init__.py ├── densenet.py ├── densenet3.py └── resnet.py ├── scripts ├── train_bake.sh ├── train_baseline.sh └── val.sh ├── train.py └── utils.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yxgeee/BAKE/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yxgeee/BAKE/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yxgeee/BAKE/HEAD/README.md -------------------------------------------------------------------------------- /bake.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yxgeee/BAKE/HEAD/bake.png -------------------------------------------------------------------------------- /imagenet/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yxgeee/BAKE/HEAD/imagenet/LICENSE -------------------------------------------------------------------------------- /imagenet/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yxgeee/BAKE/HEAD/imagenet/README.md -------------------------------------------------------------------------------- /imagenet/configs/effnet/BAKE-EN-B0_dds_8gpu.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yxgeee/BAKE/HEAD/imagenet/configs/effnet/BAKE-EN-B0_dds_8gpu.yaml -------------------------------------------------------------------------------- /imagenet/configs/effnet/EN-B0_dds_8gpu.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yxgeee/BAKE/HEAD/imagenet/configs/effnet/EN-B0_dds_8gpu.yaml -------------------------------------------------------------------------------- /imagenet/configs/mobilenet/BAKE-M-V2-W1_dds_4gpu.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yxgeee/BAKE/HEAD/imagenet/configs/mobilenet/BAKE-M-V2-W1_dds_4gpu.yaml -------------------------------------------------------------------------------- /imagenet/configs/mobilenet/M-V2-W1_dds_4gpu.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yxgeee/BAKE/HEAD/imagenet/configs/mobilenet/M-V2-W1_dds_4gpu.yaml -------------------------------------------------------------------------------- /imagenet/configs/resnest/BAKE-S-101_dds_8gpu.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yxgeee/BAKE/HEAD/imagenet/configs/resnest/BAKE-S-101_dds_8gpu.yaml -------------------------------------------------------------------------------- /imagenet/configs/resnest/BAKE-S-50_dds_8gpu.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yxgeee/BAKE/HEAD/imagenet/configs/resnest/BAKE-S-50_dds_8gpu.yaml -------------------------------------------------------------------------------- /imagenet/configs/resnest/S-101_dds_8gpu.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yxgeee/BAKE/HEAD/imagenet/configs/resnest/S-101_dds_8gpu.yaml -------------------------------------------------------------------------------- /imagenet/configs/resnest/S-50_dds_8gpu.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yxgeee/BAKE/HEAD/imagenet/configs/resnest/S-50_dds_8gpu.yaml -------------------------------------------------------------------------------- /imagenet/configs/resnet/BAKE-R-101-1x64d_dds_8gpu.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yxgeee/BAKE/HEAD/imagenet/configs/resnet/BAKE-R-101-1x64d_dds_8gpu.yaml -------------------------------------------------------------------------------- /imagenet/configs/resnet/BAKE-R-152-1x64d_dds_8gpu.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yxgeee/BAKE/HEAD/imagenet/configs/resnet/BAKE-R-152-1x64d_dds_8gpu.yaml -------------------------------------------------------------------------------- /imagenet/configs/resnet/BAKE-R-50-1x64d_dds_8gpu.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yxgeee/BAKE/HEAD/imagenet/configs/resnet/BAKE-R-50-1x64d_dds_8gpu.yaml -------------------------------------------------------------------------------- /imagenet/configs/resnet/R-101-1x64d_dds_8gpu.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yxgeee/BAKE/HEAD/imagenet/configs/resnet/R-101-1x64d_dds_8gpu.yaml -------------------------------------------------------------------------------- /imagenet/configs/resnet/R-152-1x64d_dds_8gpu.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yxgeee/BAKE/HEAD/imagenet/configs/resnet/R-152-1x64d_dds_8gpu.yaml -------------------------------------------------------------------------------- /imagenet/configs/resnet/R-50-1x64d_dds_8gpu.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yxgeee/BAKE/HEAD/imagenet/configs/resnet/R-50-1x64d_dds_8gpu.yaml -------------------------------------------------------------------------------- /imagenet/configs/resnext/BAKE-X-101-32x4d_dds_8gpu.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yxgeee/BAKE/HEAD/imagenet/configs/resnext/BAKE-X-101-32x4d_dds_8gpu.yaml -------------------------------------------------------------------------------- /imagenet/configs/resnext/BAKE-X-152-32x4d_dds_8gpu.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yxgeee/BAKE/HEAD/imagenet/configs/resnext/BAKE-X-152-32x4d_dds_8gpu.yaml -------------------------------------------------------------------------------- /imagenet/configs/resnext/X-101-32x4d_dds_8gpu.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yxgeee/BAKE/HEAD/imagenet/configs/resnext/X-101-32x4d_dds_8gpu.yaml -------------------------------------------------------------------------------- /imagenet/configs/resnext/X-152-32x4d_dds_8gpu.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yxgeee/BAKE/HEAD/imagenet/configs/resnext/X-152-32x4d_dds_8gpu.yaml -------------------------------------------------------------------------------- /imagenet/pycls/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /imagenet/pycls/core/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /imagenet/pycls/core/benchmark.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yxgeee/BAKE/HEAD/imagenet/pycls/core/benchmark.py -------------------------------------------------------------------------------- /imagenet/pycls/core/builders.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yxgeee/BAKE/HEAD/imagenet/pycls/core/builders.py -------------------------------------------------------------------------------- /imagenet/pycls/core/checkpoint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yxgeee/BAKE/HEAD/imagenet/pycls/core/checkpoint.py -------------------------------------------------------------------------------- /imagenet/pycls/core/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yxgeee/BAKE/HEAD/imagenet/pycls/core/config.py -------------------------------------------------------------------------------- /imagenet/pycls/core/distributed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yxgeee/BAKE/HEAD/imagenet/pycls/core/distributed.py -------------------------------------------------------------------------------- /imagenet/pycls/core/io.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yxgeee/BAKE/HEAD/imagenet/pycls/core/io.py -------------------------------------------------------------------------------- /imagenet/pycls/core/logging.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yxgeee/BAKE/HEAD/imagenet/pycls/core/logging.py -------------------------------------------------------------------------------- /imagenet/pycls/core/meters.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yxgeee/BAKE/HEAD/imagenet/pycls/core/meters.py -------------------------------------------------------------------------------- /imagenet/pycls/core/net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yxgeee/BAKE/HEAD/imagenet/pycls/core/net.py -------------------------------------------------------------------------------- /imagenet/pycls/core/optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yxgeee/BAKE/HEAD/imagenet/pycls/core/optimizer.py -------------------------------------------------------------------------------- /imagenet/pycls/core/plotting.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yxgeee/BAKE/HEAD/imagenet/pycls/core/plotting.py -------------------------------------------------------------------------------- /imagenet/pycls/core/timer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yxgeee/BAKE/HEAD/imagenet/pycls/core/timer.py -------------------------------------------------------------------------------- /imagenet/pycls/core/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yxgeee/BAKE/HEAD/imagenet/pycls/core/trainer.py -------------------------------------------------------------------------------- /imagenet/pycls/datasets/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /imagenet/pycls/datasets/imagenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yxgeee/BAKE/HEAD/imagenet/pycls/datasets/imagenet.py -------------------------------------------------------------------------------- /imagenet/pycls/datasets/loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yxgeee/BAKE/HEAD/imagenet/pycls/datasets/loader.py -------------------------------------------------------------------------------- /imagenet/pycls/datasets/sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yxgeee/BAKE/HEAD/imagenet/pycls/datasets/sampler.py -------------------------------------------------------------------------------- /imagenet/pycls/datasets/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yxgeee/BAKE/HEAD/imagenet/pycls/datasets/transforms.py -------------------------------------------------------------------------------- /imagenet/pycls/models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /imagenet/pycls/models/anynet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yxgeee/BAKE/HEAD/imagenet/pycls/models/anynet.py -------------------------------------------------------------------------------- /imagenet/pycls/models/effnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yxgeee/BAKE/HEAD/imagenet/pycls/models/effnet.py -------------------------------------------------------------------------------- /imagenet/pycls/models/mobilenetv2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yxgeee/BAKE/HEAD/imagenet/pycls/models/mobilenetv2.py -------------------------------------------------------------------------------- /imagenet/pycls/models/resnest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yxgeee/BAKE/HEAD/imagenet/pycls/models/resnest.py -------------------------------------------------------------------------------- /imagenet/pycls/models/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yxgeee/BAKE/HEAD/imagenet/pycls/models/resnet.py -------------------------------------------------------------------------------- /imagenet/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yxgeee/BAKE/HEAD/imagenet/requirements.txt -------------------------------------------------------------------------------- /imagenet/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yxgeee/BAKE/HEAD/imagenet/setup.py -------------------------------------------------------------------------------- /imagenet/tools/dist_test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yxgeee/BAKE/HEAD/imagenet/tools/dist_test.sh -------------------------------------------------------------------------------- /imagenet/tools/dist_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yxgeee/BAKE/HEAD/imagenet/tools/dist_train.sh -------------------------------------------------------------------------------- /imagenet/tools/slurm_test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yxgeee/BAKE/HEAD/imagenet/tools/slurm_test.sh -------------------------------------------------------------------------------- /imagenet/tools/slurm_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yxgeee/BAKE/HEAD/imagenet/tools/slurm_train.sh -------------------------------------------------------------------------------- /imagenet/tools/test_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yxgeee/BAKE/HEAD/imagenet/tools/test_net.py -------------------------------------------------------------------------------- /imagenet/tools/train_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yxgeee/BAKE/HEAD/imagenet/tools/train_net.py -------------------------------------------------------------------------------- /small_scale/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yxgeee/BAKE/HEAD/small_scale/README.md -------------------------------------------------------------------------------- /small_scale/data/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yxgeee/BAKE/HEAD/small_scale/data/README.md -------------------------------------------------------------------------------- /small_scale/data/tools/cub.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yxgeee/BAKE/HEAD/small_scale/data/tools/cub.py -------------------------------------------------------------------------------- /small_scale/data/tools/mit67.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yxgeee/BAKE/HEAD/small_scale/data/tools/mit67.py -------------------------------------------------------------------------------- /small_scale/data/tools/stanford_dogs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yxgeee/BAKE/HEAD/small_scale/data/tools/stanford_dogs.py -------------------------------------------------------------------------------- /small_scale/data/tools/tinyimagenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yxgeee/BAKE/HEAD/small_scale/data/tools/tinyimagenet.py -------------------------------------------------------------------------------- /small_scale/datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yxgeee/BAKE/HEAD/small_scale/datasets.py -------------------------------------------------------------------------------- /small_scale/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yxgeee/BAKE/HEAD/small_scale/models/__init__.py -------------------------------------------------------------------------------- /small_scale/models/densenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yxgeee/BAKE/HEAD/small_scale/models/densenet.py -------------------------------------------------------------------------------- /small_scale/models/densenet3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yxgeee/BAKE/HEAD/small_scale/models/densenet3.py -------------------------------------------------------------------------------- /small_scale/models/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yxgeee/BAKE/HEAD/small_scale/models/resnet.py -------------------------------------------------------------------------------- /small_scale/scripts/train_bake.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yxgeee/BAKE/HEAD/small_scale/scripts/train_bake.sh -------------------------------------------------------------------------------- /small_scale/scripts/train_baseline.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yxgeee/BAKE/HEAD/small_scale/scripts/train_baseline.sh -------------------------------------------------------------------------------- /small_scale/scripts/val.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yxgeee/BAKE/HEAD/small_scale/scripts/val.sh -------------------------------------------------------------------------------- /small_scale/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yxgeee/BAKE/HEAD/small_scale/train.py -------------------------------------------------------------------------------- /small_scale/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yxgeee/BAKE/HEAD/small_scale/utils.py --------------------------------------------------------------------------------