├── .gitignore ├── Adv ├── README.md ├── attack_algo.py ├── functions.py ├── models │ ├── __init__.py │ ├── modules.py │ ├── network.py │ └── utils.py ├── requirements.txt ├── scripts │ ├── test_auxbn.sh │ ├── test_bn.sh │ ├── test_saauxbn.sh │ ├── train_auxbn.sh │ ├── train_bn.sh │ └── train_saauxbn.sh ├── test.py └── train.py ├── GAN ├── README.md ├── cfg.py ├── datasets.py ├── functions.py ├── image_list_dog_and_cat.txt ├── metrics │ ├── __init__.py │ ├── fid_score_torch.py │ ├── inception_score.py │ └── inception_torch.py ├── models │ ├── __init__.py │ ├── autogan_32.py │ ├── modules.py │ ├── sngan_32.py │ └── sngan_small_128.py ├── requirements.txt ├── scripts │ ├── test_autogan_sabn_cifar10.sh │ ├── test_sngan_sabn_cifar10.sh │ ├── test_sngan_sabn_imagenet.sh │ ├── train_autogan_ccbn_cifar10.sh │ ├── train_autogan_sabn_cifar10.sh │ ├── train_sngan_ccbn_cifar10.sh │ ├── train_sngan_ccbn_imagenet.sh │ ├── train_sngan_sabn_cifar10.sh │ └── train_sngan_sabn_imagenet.sh ├── test.py ├── train.py └── utils.py ├── LICENSE ├── NAS ├── DARTS-V1.py ├── README.md ├── configs │ └── nas-benchmark │ │ ├── ImageNet-16.config │ │ ├── ImageNet16-120-split.txt │ │ ├── algos │ │ └── DARTS.config │ │ ├── cifar100-test-split.txt │ │ └── imagenet-16-120-test-split.txt ├── lib │ ├── config_utils │ │ ├── __init__.py │ │ ├── attention_args.py │ │ ├── basic_args.py │ │ ├── cls_init_args.py │ │ ├── cls_kd_args.py │ │ ├── configure_utils.py │ │ ├── pruning_args.py │ │ ├── random_baseline.py │ │ ├── search_args.py │ │ ├── search_single_args.py │ │ └── share_args.py │ ├── datasets │ │ ├── DownsampledImageNet.py │ │ ├── LandmarkDataset.py │ │ ├── SearchDatasetWrap.py │ │ ├── __init__.py │ │ ├── get_dataset_with_transform.py │ │ ├── landmark_utils │ │ │ ├── __init__.py │ │ │ └── point_meta.py │ │ └── test_utils.py │ ├── log_utils │ │ ├── __init__.py │ │ ├── logger.py │ │ ├── meter.py │ │ └── time_utils.py │ ├── models │ │ ├── SharedUtils.py │ │ ├── __init__.py │ │ ├── cell_infers │ │ │ ├── __init__.py │ │ │ ├── cells.py │ │ │ ├── nasnet_cifar.py │ │ │ └── tiny_network.py │ │ ├── cell_operations.py │ │ ├── cell_operations_ccbn.py │ │ ├── cell_operations_sabn.py │ │ ├── cell_searchs │ │ │ ├── __init__.py │ │ │ ├── genotypes.py │ │ │ ├── search_cells.py │ │ │ ├── search_cells_ccbn.py │ │ │ ├── search_cells_sabn.py │ │ │ ├── search_model_darts.py │ │ │ ├── search_model_darts_ccbn.py │ │ │ └── search_model_darts_sabn.py │ │ ├── clone_weights.py │ │ ├── initialization.py │ │ ├── model_utils.py │ │ └── norm_modules.py │ ├── nas_201_api │ │ ├── __init__.py │ │ └── api.py │ ├── procedures │ │ ├── __init__.py │ │ ├── basic_main.py │ │ ├── funcs_nasbench.py │ │ ├── optimizers.py │ │ ├── search_main.py │ │ ├── search_main_v2.py │ │ ├── simple_KD_main.py │ │ └── starts.py │ └── utils │ │ ├── __init__.py │ │ ├── affine_utils.py │ │ ├── evaluation_utils.py │ │ ├── flop_benchmark.py │ │ ├── gpu_manager.py │ │ ├── nas_utils.py │ │ └── weight_watcher.py ├── requirements.txt └── scripts │ ├── DARTS-V1-ccbn.sh │ ├── DARTS-V1-sabn.sh │ └── DARTS-V1.sh ├── NST ├── README.md ├── dataset.py ├── functions.py ├── models │ ├── __init__.py │ ├── modules.py │ └── network.py ├── requirements.txt ├── scripts │ ├── adain_net.sh │ └── saadain_net.sh └── train.py ├── README.md └── imgs ├── DARTS_e35_cifar100.png ├── DARTS_e35_imagenet100.png ├── adv_atloss.png ├── architect.png ├── clean_tloss.png ├── ct_losses.png ├── sngan_imagenet.png ├── sngan_sabn_imagenet.png ├── st_losses.png ├── style_image.png ├── val_ct_losses.png └── val_st_losses.png /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/.gitignore -------------------------------------------------------------------------------- /Adv/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/Adv/README.md -------------------------------------------------------------------------------- /Adv/attack_algo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/Adv/attack_algo.py -------------------------------------------------------------------------------- /Adv/functions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/Adv/functions.py -------------------------------------------------------------------------------- /Adv/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/Adv/models/__init__.py -------------------------------------------------------------------------------- /Adv/models/modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/Adv/models/modules.py -------------------------------------------------------------------------------- /Adv/models/network.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/Adv/models/network.py -------------------------------------------------------------------------------- /Adv/models/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/Adv/models/utils.py -------------------------------------------------------------------------------- /Adv/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/Adv/requirements.txt -------------------------------------------------------------------------------- /Adv/scripts/test_auxbn.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/Adv/scripts/test_auxbn.sh -------------------------------------------------------------------------------- /Adv/scripts/test_bn.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/Adv/scripts/test_bn.sh -------------------------------------------------------------------------------- /Adv/scripts/test_saauxbn.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/Adv/scripts/test_saauxbn.sh -------------------------------------------------------------------------------- /Adv/scripts/train_auxbn.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/Adv/scripts/train_auxbn.sh -------------------------------------------------------------------------------- /Adv/scripts/train_bn.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/Adv/scripts/train_bn.sh -------------------------------------------------------------------------------- /Adv/scripts/train_saauxbn.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/Adv/scripts/train_saauxbn.sh -------------------------------------------------------------------------------- /Adv/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/Adv/test.py -------------------------------------------------------------------------------- /Adv/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/Adv/train.py -------------------------------------------------------------------------------- /GAN/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/GAN/README.md -------------------------------------------------------------------------------- /GAN/cfg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/GAN/cfg.py -------------------------------------------------------------------------------- /GAN/datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/GAN/datasets.py -------------------------------------------------------------------------------- /GAN/functions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/GAN/functions.py -------------------------------------------------------------------------------- /GAN/image_list_dog_and_cat.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/GAN/image_list_dog_and_cat.txt -------------------------------------------------------------------------------- /GAN/metrics/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/GAN/metrics/__init__.py -------------------------------------------------------------------------------- /GAN/metrics/fid_score_torch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/GAN/metrics/fid_score_torch.py -------------------------------------------------------------------------------- /GAN/metrics/inception_score.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/GAN/metrics/inception_score.py -------------------------------------------------------------------------------- /GAN/metrics/inception_torch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/GAN/metrics/inception_torch.py -------------------------------------------------------------------------------- /GAN/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/GAN/models/__init__.py -------------------------------------------------------------------------------- /GAN/models/autogan_32.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/GAN/models/autogan_32.py -------------------------------------------------------------------------------- /GAN/models/modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/GAN/models/modules.py -------------------------------------------------------------------------------- /GAN/models/sngan_32.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/GAN/models/sngan_32.py -------------------------------------------------------------------------------- /GAN/models/sngan_small_128.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/GAN/models/sngan_small_128.py -------------------------------------------------------------------------------- /GAN/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/GAN/requirements.txt -------------------------------------------------------------------------------- /GAN/scripts/test_autogan_sabn_cifar10.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/GAN/scripts/test_autogan_sabn_cifar10.sh -------------------------------------------------------------------------------- /GAN/scripts/test_sngan_sabn_cifar10.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/GAN/scripts/test_sngan_sabn_cifar10.sh -------------------------------------------------------------------------------- /GAN/scripts/test_sngan_sabn_imagenet.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/GAN/scripts/test_sngan_sabn_imagenet.sh -------------------------------------------------------------------------------- /GAN/scripts/train_autogan_ccbn_cifar10.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/GAN/scripts/train_autogan_ccbn_cifar10.sh -------------------------------------------------------------------------------- /GAN/scripts/train_autogan_sabn_cifar10.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/GAN/scripts/train_autogan_sabn_cifar10.sh -------------------------------------------------------------------------------- /GAN/scripts/train_sngan_ccbn_cifar10.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/GAN/scripts/train_sngan_ccbn_cifar10.sh -------------------------------------------------------------------------------- /GAN/scripts/train_sngan_ccbn_imagenet.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/GAN/scripts/train_sngan_ccbn_imagenet.sh -------------------------------------------------------------------------------- /GAN/scripts/train_sngan_sabn_cifar10.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/GAN/scripts/train_sngan_sabn_cifar10.sh -------------------------------------------------------------------------------- /GAN/scripts/train_sngan_sabn_imagenet.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/GAN/scripts/train_sngan_sabn_imagenet.sh -------------------------------------------------------------------------------- /GAN/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/GAN/test.py -------------------------------------------------------------------------------- /GAN/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/GAN/train.py -------------------------------------------------------------------------------- /GAN/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/GAN/utils.py -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/LICENSE -------------------------------------------------------------------------------- /NAS/DARTS-V1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/NAS/DARTS-V1.py -------------------------------------------------------------------------------- /NAS/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/NAS/README.md -------------------------------------------------------------------------------- /NAS/configs/nas-benchmark/ImageNet-16.config: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/NAS/configs/nas-benchmark/ImageNet-16.config -------------------------------------------------------------------------------- /NAS/configs/nas-benchmark/ImageNet16-120-split.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/NAS/configs/nas-benchmark/ImageNet16-120-split.txt -------------------------------------------------------------------------------- /NAS/configs/nas-benchmark/algos/DARTS.config: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/NAS/configs/nas-benchmark/algos/DARTS.config -------------------------------------------------------------------------------- /NAS/configs/nas-benchmark/cifar100-test-split.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/NAS/configs/nas-benchmark/cifar100-test-split.txt -------------------------------------------------------------------------------- /NAS/configs/nas-benchmark/imagenet-16-120-test-split.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/NAS/configs/nas-benchmark/imagenet-16-120-test-split.txt -------------------------------------------------------------------------------- /NAS/lib/config_utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/NAS/lib/config_utils/__init__.py -------------------------------------------------------------------------------- /NAS/lib/config_utils/attention_args.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/NAS/lib/config_utils/attention_args.py -------------------------------------------------------------------------------- /NAS/lib/config_utils/basic_args.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/NAS/lib/config_utils/basic_args.py -------------------------------------------------------------------------------- /NAS/lib/config_utils/cls_init_args.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/NAS/lib/config_utils/cls_init_args.py -------------------------------------------------------------------------------- /NAS/lib/config_utils/cls_kd_args.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/NAS/lib/config_utils/cls_kd_args.py -------------------------------------------------------------------------------- /NAS/lib/config_utils/configure_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/NAS/lib/config_utils/configure_utils.py -------------------------------------------------------------------------------- /NAS/lib/config_utils/pruning_args.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/NAS/lib/config_utils/pruning_args.py -------------------------------------------------------------------------------- /NAS/lib/config_utils/random_baseline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/NAS/lib/config_utils/random_baseline.py -------------------------------------------------------------------------------- /NAS/lib/config_utils/search_args.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/NAS/lib/config_utils/search_args.py -------------------------------------------------------------------------------- /NAS/lib/config_utils/search_single_args.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/NAS/lib/config_utils/search_single_args.py -------------------------------------------------------------------------------- /NAS/lib/config_utils/share_args.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/NAS/lib/config_utils/share_args.py -------------------------------------------------------------------------------- /NAS/lib/datasets/DownsampledImageNet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/NAS/lib/datasets/DownsampledImageNet.py -------------------------------------------------------------------------------- /NAS/lib/datasets/LandmarkDataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/NAS/lib/datasets/LandmarkDataset.py -------------------------------------------------------------------------------- /NAS/lib/datasets/SearchDatasetWrap.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/NAS/lib/datasets/SearchDatasetWrap.py -------------------------------------------------------------------------------- /NAS/lib/datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/NAS/lib/datasets/__init__.py -------------------------------------------------------------------------------- /NAS/lib/datasets/get_dataset_with_transform.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/NAS/lib/datasets/get_dataset_with_transform.py -------------------------------------------------------------------------------- /NAS/lib/datasets/landmark_utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/NAS/lib/datasets/landmark_utils/__init__.py -------------------------------------------------------------------------------- /NAS/lib/datasets/landmark_utils/point_meta.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/NAS/lib/datasets/landmark_utils/point_meta.py -------------------------------------------------------------------------------- /NAS/lib/datasets/test_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/NAS/lib/datasets/test_utils.py -------------------------------------------------------------------------------- /NAS/lib/log_utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/NAS/lib/log_utils/__init__.py -------------------------------------------------------------------------------- /NAS/lib/log_utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/NAS/lib/log_utils/logger.py -------------------------------------------------------------------------------- /NAS/lib/log_utils/meter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/NAS/lib/log_utils/meter.py -------------------------------------------------------------------------------- /NAS/lib/log_utils/time_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/NAS/lib/log_utils/time_utils.py -------------------------------------------------------------------------------- /NAS/lib/models/SharedUtils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/NAS/lib/models/SharedUtils.py -------------------------------------------------------------------------------- /NAS/lib/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/NAS/lib/models/__init__.py -------------------------------------------------------------------------------- /NAS/lib/models/cell_infers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/NAS/lib/models/cell_infers/__init__.py -------------------------------------------------------------------------------- /NAS/lib/models/cell_infers/cells.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/NAS/lib/models/cell_infers/cells.py -------------------------------------------------------------------------------- /NAS/lib/models/cell_infers/nasnet_cifar.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/NAS/lib/models/cell_infers/nasnet_cifar.py -------------------------------------------------------------------------------- /NAS/lib/models/cell_infers/tiny_network.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/NAS/lib/models/cell_infers/tiny_network.py -------------------------------------------------------------------------------- /NAS/lib/models/cell_operations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/NAS/lib/models/cell_operations.py -------------------------------------------------------------------------------- /NAS/lib/models/cell_operations_ccbn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/NAS/lib/models/cell_operations_ccbn.py -------------------------------------------------------------------------------- /NAS/lib/models/cell_operations_sabn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/NAS/lib/models/cell_operations_sabn.py -------------------------------------------------------------------------------- /NAS/lib/models/cell_searchs/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/NAS/lib/models/cell_searchs/__init__.py -------------------------------------------------------------------------------- /NAS/lib/models/cell_searchs/genotypes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/NAS/lib/models/cell_searchs/genotypes.py -------------------------------------------------------------------------------- /NAS/lib/models/cell_searchs/search_cells.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/NAS/lib/models/cell_searchs/search_cells.py -------------------------------------------------------------------------------- /NAS/lib/models/cell_searchs/search_cells_ccbn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/NAS/lib/models/cell_searchs/search_cells_ccbn.py -------------------------------------------------------------------------------- /NAS/lib/models/cell_searchs/search_cells_sabn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/NAS/lib/models/cell_searchs/search_cells_sabn.py -------------------------------------------------------------------------------- /NAS/lib/models/cell_searchs/search_model_darts.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/NAS/lib/models/cell_searchs/search_model_darts.py -------------------------------------------------------------------------------- /NAS/lib/models/cell_searchs/search_model_darts_ccbn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/NAS/lib/models/cell_searchs/search_model_darts_ccbn.py -------------------------------------------------------------------------------- /NAS/lib/models/cell_searchs/search_model_darts_sabn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/NAS/lib/models/cell_searchs/search_model_darts_sabn.py -------------------------------------------------------------------------------- /NAS/lib/models/clone_weights.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/NAS/lib/models/clone_weights.py -------------------------------------------------------------------------------- /NAS/lib/models/initialization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/NAS/lib/models/initialization.py -------------------------------------------------------------------------------- /NAS/lib/models/model_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/NAS/lib/models/model_utils.py -------------------------------------------------------------------------------- /NAS/lib/models/norm_modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/NAS/lib/models/norm_modules.py -------------------------------------------------------------------------------- /NAS/lib/nas_201_api/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/NAS/lib/nas_201_api/__init__.py -------------------------------------------------------------------------------- /NAS/lib/nas_201_api/api.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/NAS/lib/nas_201_api/api.py -------------------------------------------------------------------------------- /NAS/lib/procedures/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/NAS/lib/procedures/__init__.py -------------------------------------------------------------------------------- /NAS/lib/procedures/basic_main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/NAS/lib/procedures/basic_main.py -------------------------------------------------------------------------------- /NAS/lib/procedures/funcs_nasbench.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/NAS/lib/procedures/funcs_nasbench.py -------------------------------------------------------------------------------- /NAS/lib/procedures/optimizers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/NAS/lib/procedures/optimizers.py -------------------------------------------------------------------------------- /NAS/lib/procedures/search_main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/NAS/lib/procedures/search_main.py -------------------------------------------------------------------------------- /NAS/lib/procedures/search_main_v2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/NAS/lib/procedures/search_main_v2.py -------------------------------------------------------------------------------- /NAS/lib/procedures/simple_KD_main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/NAS/lib/procedures/simple_KD_main.py -------------------------------------------------------------------------------- /NAS/lib/procedures/starts.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/NAS/lib/procedures/starts.py -------------------------------------------------------------------------------- /NAS/lib/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/NAS/lib/utils/__init__.py -------------------------------------------------------------------------------- /NAS/lib/utils/affine_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/NAS/lib/utils/affine_utils.py -------------------------------------------------------------------------------- /NAS/lib/utils/evaluation_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/NAS/lib/utils/evaluation_utils.py -------------------------------------------------------------------------------- /NAS/lib/utils/flop_benchmark.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/NAS/lib/utils/flop_benchmark.py -------------------------------------------------------------------------------- /NAS/lib/utils/gpu_manager.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/NAS/lib/utils/gpu_manager.py -------------------------------------------------------------------------------- /NAS/lib/utils/nas_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/NAS/lib/utils/nas_utils.py -------------------------------------------------------------------------------- /NAS/lib/utils/weight_watcher.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/NAS/lib/utils/weight_watcher.py -------------------------------------------------------------------------------- /NAS/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/NAS/requirements.txt -------------------------------------------------------------------------------- /NAS/scripts/DARTS-V1-ccbn.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/NAS/scripts/DARTS-V1-ccbn.sh -------------------------------------------------------------------------------- /NAS/scripts/DARTS-V1-sabn.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/NAS/scripts/DARTS-V1-sabn.sh -------------------------------------------------------------------------------- /NAS/scripts/DARTS-V1.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/NAS/scripts/DARTS-V1.sh -------------------------------------------------------------------------------- /NST/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/NST/README.md -------------------------------------------------------------------------------- /NST/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/NST/dataset.py -------------------------------------------------------------------------------- /NST/functions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/NST/functions.py -------------------------------------------------------------------------------- /NST/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/NST/models/__init__.py -------------------------------------------------------------------------------- /NST/models/modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/NST/models/modules.py -------------------------------------------------------------------------------- /NST/models/network.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/NST/models/network.py -------------------------------------------------------------------------------- /NST/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/NST/requirements.txt -------------------------------------------------------------------------------- /NST/scripts/adain_net.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/NST/scripts/adain_net.sh -------------------------------------------------------------------------------- /NST/scripts/saadain_net.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/NST/scripts/saadain_net.sh -------------------------------------------------------------------------------- /NST/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/NST/train.py -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/README.md -------------------------------------------------------------------------------- /imgs/DARTS_e35_cifar100.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/imgs/DARTS_e35_cifar100.png -------------------------------------------------------------------------------- /imgs/DARTS_e35_imagenet100.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/imgs/DARTS_e35_imagenet100.png -------------------------------------------------------------------------------- /imgs/adv_atloss.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/imgs/adv_atloss.png -------------------------------------------------------------------------------- /imgs/architect.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/imgs/architect.png -------------------------------------------------------------------------------- /imgs/clean_tloss.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/imgs/clean_tloss.png -------------------------------------------------------------------------------- /imgs/ct_losses.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/imgs/ct_losses.png -------------------------------------------------------------------------------- /imgs/sngan_imagenet.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/imgs/sngan_imagenet.png -------------------------------------------------------------------------------- /imgs/sngan_sabn_imagenet.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/imgs/sngan_sabn_imagenet.png -------------------------------------------------------------------------------- /imgs/st_losses.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/imgs/st_losses.png -------------------------------------------------------------------------------- /imgs/style_image.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/imgs/style_image.png -------------------------------------------------------------------------------- /imgs/val_ct_losses.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/imgs/val_ct_losses.png -------------------------------------------------------------------------------- /imgs/val_st_losses.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/Sandwich-Batch-Normalization/HEAD/imgs/val_st_losses.png --------------------------------------------------------------------------------