├── LICENSE ├── README.md ├── args.py ├── configs ├── OE │ └── oe-baseline.yaml ├── estimate_loss │ └── estimate_loss_cifar10.yaml ├── parser.py └── smallscale │ ├── densenet_cifar10.yaml │ ├── densenet_cifar100.yaml │ ├── densenet_cifar100_UMAP.yaml │ └── densenet_cifar10_UMAP.yaml ├── data ├── OOD.py ├── __init__.py ├── bigcifar.py ├── cifar.py ├── cifar100.py ├── imagenet.py ├── imagenet_oe.py ├── mnist.py ├── svhn.py ├── tinyimagenet.py └── utils.py ├── estimate_loss.py ├── feature_stat ├── CIFAR-100_densenet_feat_stat.npy └── CIFAR-10_densenet_feat_stat.npy ├── figures ├── ICML_UM_poster.png └── framework_overview.jpg ├── main.py ├── main_OE.py ├── models ├── __init__.py ├── densenet.py ├── frankle.py ├── resnet.py ├── resnet_cifar.py └── wideresnet.py ├── requirements.txt ├── runs ├── densenet_cifar100_best.state ├── densenet_cifar100_last.state ├── densenet_cifar10_best.state └── densenet_cifar10_last.state ├── trainers └── default.py └── utils ├── __init__.py ├── ash.py ├── bn_type.py ├── builder.py ├── conv_type.py ├── custom_loss.py ├── eval_utils.py ├── get_metrics.py ├── get_scores.py ├── linear_type.py ├── logging.py ├── mahalanobis_lib.py ├── net_utils.py ├── neural_linear_opt.py ├── profiling.py ├── schedulers.py └── utils_awp.py /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZFancy/Unleashing-Mask/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZFancy/Unleashing-Mask/HEAD/README.md -------------------------------------------------------------------------------- /args.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZFancy/Unleashing-Mask/HEAD/args.py -------------------------------------------------------------------------------- /configs/OE/oe-baseline.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZFancy/Unleashing-Mask/HEAD/configs/OE/oe-baseline.yaml -------------------------------------------------------------------------------- /configs/estimate_loss/estimate_loss_cifar10.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZFancy/Unleashing-Mask/HEAD/configs/estimate_loss/estimate_loss_cifar10.yaml -------------------------------------------------------------------------------- /configs/parser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZFancy/Unleashing-Mask/HEAD/configs/parser.py -------------------------------------------------------------------------------- /configs/smallscale/densenet_cifar10.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZFancy/Unleashing-Mask/HEAD/configs/smallscale/densenet_cifar10.yaml -------------------------------------------------------------------------------- /configs/smallscale/densenet_cifar100.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZFancy/Unleashing-Mask/HEAD/configs/smallscale/densenet_cifar100.yaml -------------------------------------------------------------------------------- /configs/smallscale/densenet_cifar100_UMAP.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZFancy/Unleashing-Mask/HEAD/configs/smallscale/densenet_cifar100_UMAP.yaml -------------------------------------------------------------------------------- /configs/smallscale/densenet_cifar10_UMAP.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZFancy/Unleashing-Mask/HEAD/configs/smallscale/densenet_cifar10_UMAP.yaml -------------------------------------------------------------------------------- /data/OOD.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZFancy/Unleashing-Mask/HEAD/data/OOD.py -------------------------------------------------------------------------------- /data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZFancy/Unleashing-Mask/HEAD/data/__init__.py -------------------------------------------------------------------------------- /data/bigcifar.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZFancy/Unleashing-Mask/HEAD/data/bigcifar.py -------------------------------------------------------------------------------- /data/cifar.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZFancy/Unleashing-Mask/HEAD/data/cifar.py -------------------------------------------------------------------------------- /data/cifar100.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZFancy/Unleashing-Mask/HEAD/data/cifar100.py -------------------------------------------------------------------------------- /data/imagenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZFancy/Unleashing-Mask/HEAD/data/imagenet.py -------------------------------------------------------------------------------- /data/imagenet_oe.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZFancy/Unleashing-Mask/HEAD/data/imagenet_oe.py -------------------------------------------------------------------------------- /data/mnist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZFancy/Unleashing-Mask/HEAD/data/mnist.py -------------------------------------------------------------------------------- /data/svhn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZFancy/Unleashing-Mask/HEAD/data/svhn.py -------------------------------------------------------------------------------- /data/tinyimagenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZFancy/Unleashing-Mask/HEAD/data/tinyimagenet.py -------------------------------------------------------------------------------- /data/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZFancy/Unleashing-Mask/HEAD/data/utils.py -------------------------------------------------------------------------------- /estimate_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZFancy/Unleashing-Mask/HEAD/estimate_loss.py -------------------------------------------------------------------------------- /feature_stat/CIFAR-100_densenet_feat_stat.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZFancy/Unleashing-Mask/HEAD/feature_stat/CIFAR-100_densenet_feat_stat.npy -------------------------------------------------------------------------------- /feature_stat/CIFAR-10_densenet_feat_stat.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZFancy/Unleashing-Mask/HEAD/feature_stat/CIFAR-10_densenet_feat_stat.npy -------------------------------------------------------------------------------- /figures/ICML_UM_poster.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZFancy/Unleashing-Mask/HEAD/figures/ICML_UM_poster.png -------------------------------------------------------------------------------- /figures/framework_overview.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZFancy/Unleashing-Mask/HEAD/figures/framework_overview.jpg -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZFancy/Unleashing-Mask/HEAD/main.py -------------------------------------------------------------------------------- /main_OE.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZFancy/Unleashing-Mask/HEAD/main_OE.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZFancy/Unleashing-Mask/HEAD/models/__init__.py -------------------------------------------------------------------------------- /models/densenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZFancy/Unleashing-Mask/HEAD/models/densenet.py -------------------------------------------------------------------------------- /models/frankle.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZFancy/Unleashing-Mask/HEAD/models/frankle.py -------------------------------------------------------------------------------- /models/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZFancy/Unleashing-Mask/HEAD/models/resnet.py -------------------------------------------------------------------------------- /models/resnet_cifar.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZFancy/Unleashing-Mask/HEAD/models/resnet_cifar.py -------------------------------------------------------------------------------- /models/wideresnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZFancy/Unleashing-Mask/HEAD/models/wideresnet.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZFancy/Unleashing-Mask/HEAD/requirements.txt -------------------------------------------------------------------------------- /runs/densenet_cifar100_best.state: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZFancy/Unleashing-Mask/HEAD/runs/densenet_cifar100_best.state -------------------------------------------------------------------------------- /runs/densenet_cifar100_last.state: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZFancy/Unleashing-Mask/HEAD/runs/densenet_cifar100_last.state -------------------------------------------------------------------------------- /runs/densenet_cifar10_best.state: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZFancy/Unleashing-Mask/HEAD/runs/densenet_cifar10_best.state -------------------------------------------------------------------------------- /runs/densenet_cifar10_last.state: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZFancy/Unleashing-Mask/HEAD/runs/densenet_cifar10_last.state -------------------------------------------------------------------------------- /trainers/default.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZFancy/Unleashing-Mask/HEAD/trainers/default.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/ash.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZFancy/Unleashing-Mask/HEAD/utils/ash.py -------------------------------------------------------------------------------- /utils/bn_type.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZFancy/Unleashing-Mask/HEAD/utils/bn_type.py -------------------------------------------------------------------------------- /utils/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZFancy/Unleashing-Mask/HEAD/utils/builder.py -------------------------------------------------------------------------------- /utils/conv_type.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZFancy/Unleashing-Mask/HEAD/utils/conv_type.py -------------------------------------------------------------------------------- /utils/custom_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZFancy/Unleashing-Mask/HEAD/utils/custom_loss.py -------------------------------------------------------------------------------- /utils/eval_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZFancy/Unleashing-Mask/HEAD/utils/eval_utils.py -------------------------------------------------------------------------------- /utils/get_metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZFancy/Unleashing-Mask/HEAD/utils/get_metrics.py -------------------------------------------------------------------------------- /utils/get_scores.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZFancy/Unleashing-Mask/HEAD/utils/get_scores.py -------------------------------------------------------------------------------- /utils/linear_type.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZFancy/Unleashing-Mask/HEAD/utils/linear_type.py -------------------------------------------------------------------------------- /utils/logging.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZFancy/Unleashing-Mask/HEAD/utils/logging.py -------------------------------------------------------------------------------- /utils/mahalanobis_lib.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZFancy/Unleashing-Mask/HEAD/utils/mahalanobis_lib.py -------------------------------------------------------------------------------- /utils/net_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZFancy/Unleashing-Mask/HEAD/utils/net_utils.py -------------------------------------------------------------------------------- /utils/neural_linear_opt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZFancy/Unleashing-Mask/HEAD/utils/neural_linear_opt.py -------------------------------------------------------------------------------- /utils/profiling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZFancy/Unleashing-Mask/HEAD/utils/profiling.py -------------------------------------------------------------------------------- /utils/schedulers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZFancy/Unleashing-Mask/HEAD/utils/schedulers.py -------------------------------------------------------------------------------- /utils/utils_awp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZFancy/Unleashing-Mask/HEAD/utils/utils_awp.py --------------------------------------------------------------------------------