├── DAB_main_LICENSE.txt ├── README.md ├── datasets ├── __init__.py ├── base.py ├── cifar.py ├── imagenet.py ├── ood.py ├── resnet_preprocessing.py └── svhn.py ├── hparams ├── cifar_hparams.py └── imagenet_hparams.py ├── layers ├── __init__.py ├── dab_normal.py ├── dab_normal_diag_covariance.py └── dab_normal_full_covariance.py ├── models ├── __init__.py ├── pretrained_resnet50_dab.py └── wide_resnet_dab.py ├── requirements.txt ├── run_cifar.py ├── run_imagenet.py ├── synthetic_regression_demo.py └── utils ├── metrics_utils.py └── schedules.py /DAB_main_LICENSE.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ifiaposto/Distance_Aware_Bottleneck/HEAD/DAB_main_LICENSE.txt -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ifiaposto/Distance_Aware_Bottleneck/HEAD/README.md -------------------------------------------------------------------------------- /datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ifiaposto/Distance_Aware_Bottleneck/HEAD/datasets/__init__.py -------------------------------------------------------------------------------- /datasets/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ifiaposto/Distance_Aware_Bottleneck/HEAD/datasets/base.py -------------------------------------------------------------------------------- /datasets/cifar.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ifiaposto/Distance_Aware_Bottleneck/HEAD/datasets/cifar.py -------------------------------------------------------------------------------- /datasets/imagenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ifiaposto/Distance_Aware_Bottleneck/HEAD/datasets/imagenet.py -------------------------------------------------------------------------------- /datasets/ood.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ifiaposto/Distance_Aware_Bottleneck/HEAD/datasets/ood.py -------------------------------------------------------------------------------- /datasets/resnet_preprocessing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ifiaposto/Distance_Aware_Bottleneck/HEAD/datasets/resnet_preprocessing.py -------------------------------------------------------------------------------- /datasets/svhn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ifiaposto/Distance_Aware_Bottleneck/HEAD/datasets/svhn.py -------------------------------------------------------------------------------- /hparams/cifar_hparams.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ifiaposto/Distance_Aware_Bottleneck/HEAD/hparams/cifar_hparams.py -------------------------------------------------------------------------------- /hparams/imagenet_hparams.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ifiaposto/Distance_Aware_Bottleneck/HEAD/hparams/imagenet_hparams.py -------------------------------------------------------------------------------- /layers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ifiaposto/Distance_Aware_Bottleneck/HEAD/layers/__init__.py -------------------------------------------------------------------------------- /layers/dab_normal.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ifiaposto/Distance_Aware_Bottleneck/HEAD/layers/dab_normal.py -------------------------------------------------------------------------------- /layers/dab_normal_diag_covariance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ifiaposto/Distance_Aware_Bottleneck/HEAD/layers/dab_normal_diag_covariance.py -------------------------------------------------------------------------------- /layers/dab_normal_full_covariance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ifiaposto/Distance_Aware_Bottleneck/HEAD/layers/dab_normal_full_covariance.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ifiaposto/Distance_Aware_Bottleneck/HEAD/models/__init__.py -------------------------------------------------------------------------------- /models/pretrained_resnet50_dab.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ifiaposto/Distance_Aware_Bottleneck/HEAD/models/pretrained_resnet50_dab.py -------------------------------------------------------------------------------- /models/wide_resnet_dab.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ifiaposto/Distance_Aware_Bottleneck/HEAD/models/wide_resnet_dab.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ifiaposto/Distance_Aware_Bottleneck/HEAD/requirements.txt -------------------------------------------------------------------------------- /run_cifar.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ifiaposto/Distance_Aware_Bottleneck/HEAD/run_cifar.py -------------------------------------------------------------------------------- /run_imagenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ifiaposto/Distance_Aware_Bottleneck/HEAD/run_imagenet.py -------------------------------------------------------------------------------- /synthetic_regression_demo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ifiaposto/Distance_Aware_Bottleneck/HEAD/synthetic_regression_demo.py -------------------------------------------------------------------------------- /utils/metrics_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ifiaposto/Distance_Aware_Bottleneck/HEAD/utils/metrics_utils.py -------------------------------------------------------------------------------- /utils/schedules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ifiaposto/Distance_Aware_Bottleneck/HEAD/utils/schedules.py --------------------------------------------------------------------------------