├── .gitignore ├── README.md ├── args.py ├── compute_flops.py ├── data ├── imagenet.py └── imagenet_dali.py ├── images ├── balance.jpg └── framework.jpg ├── models ├── __init__.py ├── bn_type.py ├── conv_type │ ├── __init__.py │ ├── base_conv.py │ ├── masker.py │ ├── regrow_uniform_block.py │ └── regrow_uniform_block_v2.py ├── imagenet_resnet.py └── mobilenet_v1.py ├── pruning_train.py ├── scripts └── generate_scripts.py └── utils ├── builder.py ├── data_utils.py ├── engine ├── __init__.py └── epoch.py ├── net_utils.py └── prune_criterion.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JingyangXiang/SUBP/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JingyangXiang/SUBP/HEAD/README.md -------------------------------------------------------------------------------- /args.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JingyangXiang/SUBP/HEAD/args.py -------------------------------------------------------------------------------- /compute_flops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JingyangXiang/SUBP/HEAD/compute_flops.py -------------------------------------------------------------------------------- /data/imagenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JingyangXiang/SUBP/HEAD/data/imagenet.py -------------------------------------------------------------------------------- /data/imagenet_dali.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JingyangXiang/SUBP/HEAD/data/imagenet_dali.py -------------------------------------------------------------------------------- /images/balance.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JingyangXiang/SUBP/HEAD/images/balance.jpg -------------------------------------------------------------------------------- /images/framework.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JingyangXiang/SUBP/HEAD/images/framework.jpg -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JingyangXiang/SUBP/HEAD/models/__init__.py -------------------------------------------------------------------------------- /models/bn_type.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JingyangXiang/SUBP/HEAD/models/bn_type.py -------------------------------------------------------------------------------- /models/conv_type/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JingyangXiang/SUBP/HEAD/models/conv_type/__init__.py -------------------------------------------------------------------------------- /models/conv_type/base_conv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JingyangXiang/SUBP/HEAD/models/conv_type/base_conv.py -------------------------------------------------------------------------------- /models/conv_type/masker.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JingyangXiang/SUBP/HEAD/models/conv_type/masker.py -------------------------------------------------------------------------------- /models/conv_type/regrow_uniform_block.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JingyangXiang/SUBP/HEAD/models/conv_type/regrow_uniform_block.py -------------------------------------------------------------------------------- /models/conv_type/regrow_uniform_block_v2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JingyangXiang/SUBP/HEAD/models/conv_type/regrow_uniform_block_v2.py -------------------------------------------------------------------------------- /models/imagenet_resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JingyangXiang/SUBP/HEAD/models/imagenet_resnet.py -------------------------------------------------------------------------------- /models/mobilenet_v1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JingyangXiang/SUBP/HEAD/models/mobilenet_v1.py -------------------------------------------------------------------------------- /pruning_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JingyangXiang/SUBP/HEAD/pruning_train.py -------------------------------------------------------------------------------- /scripts/generate_scripts.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JingyangXiang/SUBP/HEAD/scripts/generate_scripts.py -------------------------------------------------------------------------------- /utils/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JingyangXiang/SUBP/HEAD/utils/builder.py -------------------------------------------------------------------------------- /utils/data_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JingyangXiang/SUBP/HEAD/utils/data_utils.py -------------------------------------------------------------------------------- /utils/engine/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JingyangXiang/SUBP/HEAD/utils/engine/__init__.py -------------------------------------------------------------------------------- /utils/engine/epoch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JingyangXiang/SUBP/HEAD/utils/engine/epoch.py -------------------------------------------------------------------------------- /utils/net_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JingyangXiang/SUBP/HEAD/utils/net_utils.py -------------------------------------------------------------------------------- /utils/prune_criterion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JingyangXiang/SUBP/HEAD/utils/prune_criterion.py --------------------------------------------------------------------------------