├── README.md ├── args.py ├── configs ├── parser.py ├── resnet32-cifar10-pr0.1.yaml ├── resnet32-cifar100-pr0.1.yaml ├── vgg19-cifar10-pr0.1.yaml └── vgg19-cifar100-pr0.1.yaml ├── data ├── __init__.py ├── cifar.py └── mnist.py ├── main.py ├── models ├── __init__.py ├── frankle.py ├── resnet.py └── vgg_bn.py ├── trainers └── default.py └── utils ├── bn_type.py ├── builder.py ├── conv_type.py ├── eval_utils.py ├── logging.py ├── net_utils.py └── schedulers.py /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/x-zho14/ProbMask-official/HEAD/README.md -------------------------------------------------------------------------------- /args.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/x-zho14/ProbMask-official/HEAD/args.py -------------------------------------------------------------------------------- /configs/parser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/x-zho14/ProbMask-official/HEAD/configs/parser.py -------------------------------------------------------------------------------- /configs/resnet32-cifar10-pr0.1.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/x-zho14/ProbMask-official/HEAD/configs/resnet32-cifar10-pr0.1.yaml -------------------------------------------------------------------------------- /configs/resnet32-cifar100-pr0.1.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/x-zho14/ProbMask-official/HEAD/configs/resnet32-cifar100-pr0.1.yaml -------------------------------------------------------------------------------- /configs/vgg19-cifar10-pr0.1.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/x-zho14/ProbMask-official/HEAD/configs/vgg19-cifar10-pr0.1.yaml -------------------------------------------------------------------------------- /configs/vgg19-cifar100-pr0.1.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/x-zho14/ProbMask-official/HEAD/configs/vgg19-cifar100-pr0.1.yaml -------------------------------------------------------------------------------- /data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/x-zho14/ProbMask-official/HEAD/data/__init__.py -------------------------------------------------------------------------------- /data/cifar.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/x-zho14/ProbMask-official/HEAD/data/cifar.py -------------------------------------------------------------------------------- /data/mnist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/x-zho14/ProbMask-official/HEAD/data/mnist.py -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/x-zho14/ProbMask-official/HEAD/main.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/x-zho14/ProbMask-official/HEAD/models/__init__.py -------------------------------------------------------------------------------- /models/frankle.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/x-zho14/ProbMask-official/HEAD/models/frankle.py -------------------------------------------------------------------------------- /models/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/x-zho14/ProbMask-official/HEAD/models/resnet.py -------------------------------------------------------------------------------- /models/vgg_bn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/x-zho14/ProbMask-official/HEAD/models/vgg_bn.py -------------------------------------------------------------------------------- /trainers/default.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/x-zho14/ProbMask-official/HEAD/trainers/default.py -------------------------------------------------------------------------------- /utils/bn_type.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/x-zho14/ProbMask-official/HEAD/utils/bn_type.py -------------------------------------------------------------------------------- /utils/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/x-zho14/ProbMask-official/HEAD/utils/builder.py -------------------------------------------------------------------------------- /utils/conv_type.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/x-zho14/ProbMask-official/HEAD/utils/conv_type.py -------------------------------------------------------------------------------- /utils/eval_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/x-zho14/ProbMask-official/HEAD/utils/eval_utils.py -------------------------------------------------------------------------------- /utils/logging.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/x-zho14/ProbMask-official/HEAD/utils/logging.py -------------------------------------------------------------------------------- /utils/net_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/x-zho14/ProbMask-official/HEAD/utils/net_utils.py -------------------------------------------------------------------------------- /utils/schedulers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/x-zho14/ProbMask-official/HEAD/utils/schedulers.py --------------------------------------------------------------------------------