├── .gitignore ├── LICENSE ├── README.md ├── examples ├── support_model │ ├── classification.py │ ├── detection.py │ └── segmentation.py ├── torchpruner │ ├── model_tools_exmple.py │ ├── prune_and_recovery.py │ ├── prune_by_class.py │ └── prune_by_name.py └── torchslim │ ├── imagenet │ ├── acnet.py │ ├── imagenet_train.py │ ├── prune.py │ └── prune_mobilenet.py │ └── pytorch-cifar │ ├── README.md │ ├── acnet.py │ ├── main.py │ ├── models │ ├── __init__.py │ ├── densenet.py │ ├── dla.py │ ├── dla_simple.py │ ├── dpn.py │ ├── efficientnet.py │ ├── googlenet.py │ ├── lenet.py │ ├── mobilenet.py │ ├── mobilenetv2.py │ ├── pnasnet.py │ ├── preact_resnet.py │ ├── regnet.py │ ├── resnet.py │ ├── resnext.py │ ├── senet.py │ ├── shufflenet.py │ ├── shufflenetv2.py │ ├── unet.py │ └── vgg.py │ ├── prune.py │ ├── prune_csgd.py │ ├── qat.py │ └── utils.py ├── setup.py ├── torchpruner ├── DOCUMENT.md ├── README.md ├── __init__.py ├── function_module.py ├── graph.py ├── init │ ├── __init__.py │ ├── init.py │ ├── module_pruner_regist.py │ └── onnx_op_regist.py ├── mask_utils.py ├── model_pruner.py ├── model_tools.py ├── module_pruner │ ├── __init__.py │ ├── prune_function.py │ └── pruners.py ├── operator │ ├── __init__.py │ ├── onnx_operator.py │ └── operator.py └── register.py └── torchslim ├── DOCUMENT.md ├── README.md ├── __init__.py ├── modules ├── __init__.py ├── acb_corner_rep_module.py ├── acnet_rep_modules.py ├── base_rep_module.py └── cnc_rep_module.py ├── pruning ├── __init__.py ├── csgd.py └── resrep.py ├── quantizing ├── __init__.py ├── qat.py ├── qat_tools.py └── quantizer_test.py ├── reparameter ├── __init__.py └── reparam.py └── slim_solver.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THU-MIG/torch-model-compression/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THU-MIG/torch-model-compression/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THU-MIG/torch-model-compression/HEAD/README.md -------------------------------------------------------------------------------- /examples/support_model/classification.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THU-MIG/torch-model-compression/HEAD/examples/support_model/classification.py -------------------------------------------------------------------------------- /examples/support_model/detection.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THU-MIG/torch-model-compression/HEAD/examples/support_model/detection.py -------------------------------------------------------------------------------- /examples/support_model/segmentation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THU-MIG/torch-model-compression/HEAD/examples/support_model/segmentation.py -------------------------------------------------------------------------------- /examples/torchpruner/model_tools_exmple.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THU-MIG/torch-model-compression/HEAD/examples/torchpruner/model_tools_exmple.py -------------------------------------------------------------------------------- /examples/torchpruner/prune_and_recovery.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THU-MIG/torch-model-compression/HEAD/examples/torchpruner/prune_and_recovery.py -------------------------------------------------------------------------------- /examples/torchpruner/prune_by_class.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THU-MIG/torch-model-compression/HEAD/examples/torchpruner/prune_by_class.py -------------------------------------------------------------------------------- /examples/torchpruner/prune_by_name.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THU-MIG/torch-model-compression/HEAD/examples/torchpruner/prune_by_name.py -------------------------------------------------------------------------------- /examples/torchslim/imagenet/acnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THU-MIG/torch-model-compression/HEAD/examples/torchslim/imagenet/acnet.py -------------------------------------------------------------------------------- /examples/torchslim/imagenet/imagenet_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THU-MIG/torch-model-compression/HEAD/examples/torchslim/imagenet/imagenet_train.py -------------------------------------------------------------------------------- /examples/torchslim/imagenet/prune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THU-MIG/torch-model-compression/HEAD/examples/torchslim/imagenet/prune.py -------------------------------------------------------------------------------- /examples/torchslim/imagenet/prune_mobilenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THU-MIG/torch-model-compression/HEAD/examples/torchslim/imagenet/prune_mobilenet.py -------------------------------------------------------------------------------- /examples/torchslim/pytorch-cifar/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THU-MIG/torch-model-compression/HEAD/examples/torchslim/pytorch-cifar/README.md -------------------------------------------------------------------------------- /examples/torchslim/pytorch-cifar/acnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THU-MIG/torch-model-compression/HEAD/examples/torchslim/pytorch-cifar/acnet.py -------------------------------------------------------------------------------- /examples/torchslim/pytorch-cifar/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THU-MIG/torch-model-compression/HEAD/examples/torchslim/pytorch-cifar/main.py -------------------------------------------------------------------------------- /examples/torchslim/pytorch-cifar/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THU-MIG/torch-model-compression/HEAD/examples/torchslim/pytorch-cifar/models/__init__.py -------------------------------------------------------------------------------- /examples/torchslim/pytorch-cifar/models/densenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THU-MIG/torch-model-compression/HEAD/examples/torchslim/pytorch-cifar/models/densenet.py -------------------------------------------------------------------------------- /examples/torchslim/pytorch-cifar/models/dla.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THU-MIG/torch-model-compression/HEAD/examples/torchslim/pytorch-cifar/models/dla.py -------------------------------------------------------------------------------- /examples/torchslim/pytorch-cifar/models/dla_simple.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THU-MIG/torch-model-compression/HEAD/examples/torchslim/pytorch-cifar/models/dla_simple.py -------------------------------------------------------------------------------- /examples/torchslim/pytorch-cifar/models/dpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THU-MIG/torch-model-compression/HEAD/examples/torchslim/pytorch-cifar/models/dpn.py -------------------------------------------------------------------------------- /examples/torchslim/pytorch-cifar/models/efficientnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THU-MIG/torch-model-compression/HEAD/examples/torchslim/pytorch-cifar/models/efficientnet.py -------------------------------------------------------------------------------- /examples/torchslim/pytorch-cifar/models/googlenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THU-MIG/torch-model-compression/HEAD/examples/torchslim/pytorch-cifar/models/googlenet.py -------------------------------------------------------------------------------- /examples/torchslim/pytorch-cifar/models/lenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THU-MIG/torch-model-compression/HEAD/examples/torchslim/pytorch-cifar/models/lenet.py -------------------------------------------------------------------------------- /examples/torchslim/pytorch-cifar/models/mobilenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THU-MIG/torch-model-compression/HEAD/examples/torchslim/pytorch-cifar/models/mobilenet.py -------------------------------------------------------------------------------- /examples/torchslim/pytorch-cifar/models/mobilenetv2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THU-MIG/torch-model-compression/HEAD/examples/torchslim/pytorch-cifar/models/mobilenetv2.py -------------------------------------------------------------------------------- /examples/torchslim/pytorch-cifar/models/pnasnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THU-MIG/torch-model-compression/HEAD/examples/torchslim/pytorch-cifar/models/pnasnet.py -------------------------------------------------------------------------------- /examples/torchslim/pytorch-cifar/models/preact_resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THU-MIG/torch-model-compression/HEAD/examples/torchslim/pytorch-cifar/models/preact_resnet.py -------------------------------------------------------------------------------- /examples/torchslim/pytorch-cifar/models/regnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THU-MIG/torch-model-compression/HEAD/examples/torchslim/pytorch-cifar/models/regnet.py -------------------------------------------------------------------------------- /examples/torchslim/pytorch-cifar/models/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THU-MIG/torch-model-compression/HEAD/examples/torchslim/pytorch-cifar/models/resnet.py -------------------------------------------------------------------------------- /examples/torchslim/pytorch-cifar/models/resnext.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THU-MIG/torch-model-compression/HEAD/examples/torchslim/pytorch-cifar/models/resnext.py -------------------------------------------------------------------------------- /examples/torchslim/pytorch-cifar/models/senet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THU-MIG/torch-model-compression/HEAD/examples/torchslim/pytorch-cifar/models/senet.py -------------------------------------------------------------------------------- /examples/torchslim/pytorch-cifar/models/shufflenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THU-MIG/torch-model-compression/HEAD/examples/torchslim/pytorch-cifar/models/shufflenet.py -------------------------------------------------------------------------------- /examples/torchslim/pytorch-cifar/models/shufflenetv2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THU-MIG/torch-model-compression/HEAD/examples/torchslim/pytorch-cifar/models/shufflenetv2.py -------------------------------------------------------------------------------- /examples/torchslim/pytorch-cifar/models/unet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THU-MIG/torch-model-compression/HEAD/examples/torchslim/pytorch-cifar/models/unet.py -------------------------------------------------------------------------------- /examples/torchslim/pytorch-cifar/models/vgg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THU-MIG/torch-model-compression/HEAD/examples/torchslim/pytorch-cifar/models/vgg.py -------------------------------------------------------------------------------- /examples/torchslim/pytorch-cifar/prune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THU-MIG/torch-model-compression/HEAD/examples/torchslim/pytorch-cifar/prune.py -------------------------------------------------------------------------------- /examples/torchslim/pytorch-cifar/prune_csgd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THU-MIG/torch-model-compression/HEAD/examples/torchslim/pytorch-cifar/prune_csgd.py -------------------------------------------------------------------------------- /examples/torchslim/pytorch-cifar/qat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THU-MIG/torch-model-compression/HEAD/examples/torchslim/pytorch-cifar/qat.py -------------------------------------------------------------------------------- /examples/torchslim/pytorch-cifar/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THU-MIG/torch-model-compression/HEAD/examples/torchslim/pytorch-cifar/utils.py -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THU-MIG/torch-model-compression/HEAD/setup.py -------------------------------------------------------------------------------- /torchpruner/DOCUMENT.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THU-MIG/torch-model-compression/HEAD/torchpruner/DOCUMENT.md -------------------------------------------------------------------------------- /torchpruner/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THU-MIG/torch-model-compression/HEAD/torchpruner/README.md -------------------------------------------------------------------------------- /torchpruner/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THU-MIG/torch-model-compression/HEAD/torchpruner/__init__.py -------------------------------------------------------------------------------- /torchpruner/function_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THU-MIG/torch-model-compression/HEAD/torchpruner/function_module.py -------------------------------------------------------------------------------- /torchpruner/graph.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THU-MIG/torch-model-compression/HEAD/torchpruner/graph.py -------------------------------------------------------------------------------- /torchpruner/init/__init__.py: -------------------------------------------------------------------------------- 1 | from . import init 2 | -------------------------------------------------------------------------------- /torchpruner/init/init.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THU-MIG/torch-model-compression/HEAD/torchpruner/init/init.py -------------------------------------------------------------------------------- /torchpruner/init/module_pruner_regist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THU-MIG/torch-model-compression/HEAD/torchpruner/init/module_pruner_regist.py -------------------------------------------------------------------------------- /torchpruner/init/onnx_op_regist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THU-MIG/torch-model-compression/HEAD/torchpruner/init/onnx_op_regist.py -------------------------------------------------------------------------------- /torchpruner/mask_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THU-MIG/torch-model-compression/HEAD/torchpruner/mask_utils.py -------------------------------------------------------------------------------- /torchpruner/model_pruner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THU-MIG/torch-model-compression/HEAD/torchpruner/model_pruner.py -------------------------------------------------------------------------------- /torchpruner/model_tools.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THU-MIG/torch-model-compression/HEAD/torchpruner/model_tools.py -------------------------------------------------------------------------------- /torchpruner/module_pruner/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THU-MIG/torch-model-compression/HEAD/torchpruner/module_pruner/__init__.py -------------------------------------------------------------------------------- /torchpruner/module_pruner/prune_function.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THU-MIG/torch-model-compression/HEAD/torchpruner/module_pruner/prune_function.py -------------------------------------------------------------------------------- /torchpruner/module_pruner/pruners.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THU-MIG/torch-model-compression/HEAD/torchpruner/module_pruner/pruners.py -------------------------------------------------------------------------------- /torchpruner/operator/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THU-MIG/torch-model-compression/HEAD/torchpruner/operator/__init__.py -------------------------------------------------------------------------------- /torchpruner/operator/onnx_operator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THU-MIG/torch-model-compression/HEAD/torchpruner/operator/onnx_operator.py -------------------------------------------------------------------------------- /torchpruner/operator/operator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THU-MIG/torch-model-compression/HEAD/torchpruner/operator/operator.py -------------------------------------------------------------------------------- /torchpruner/register.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THU-MIG/torch-model-compression/HEAD/torchpruner/register.py -------------------------------------------------------------------------------- /torchslim/DOCUMENT.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THU-MIG/torch-model-compression/HEAD/torchslim/DOCUMENT.md -------------------------------------------------------------------------------- /torchslim/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THU-MIG/torch-model-compression/HEAD/torchslim/README.md -------------------------------------------------------------------------------- /torchslim/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THU-MIG/torch-model-compression/HEAD/torchslim/__init__.py -------------------------------------------------------------------------------- /torchslim/modules/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THU-MIG/torch-model-compression/HEAD/torchslim/modules/__init__.py -------------------------------------------------------------------------------- /torchslim/modules/acb_corner_rep_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THU-MIG/torch-model-compression/HEAD/torchslim/modules/acb_corner_rep_module.py -------------------------------------------------------------------------------- /torchslim/modules/acnet_rep_modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THU-MIG/torch-model-compression/HEAD/torchslim/modules/acnet_rep_modules.py -------------------------------------------------------------------------------- /torchslim/modules/base_rep_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THU-MIG/torch-model-compression/HEAD/torchslim/modules/base_rep_module.py -------------------------------------------------------------------------------- /torchslim/modules/cnc_rep_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THU-MIG/torch-model-compression/HEAD/torchslim/modules/cnc_rep_module.py -------------------------------------------------------------------------------- /torchslim/pruning/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /torchslim/pruning/csgd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THU-MIG/torch-model-compression/HEAD/torchslim/pruning/csgd.py -------------------------------------------------------------------------------- /torchslim/pruning/resrep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THU-MIG/torch-model-compression/HEAD/torchslim/pruning/resrep.py -------------------------------------------------------------------------------- /torchslim/quantizing/__init__.py: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /torchslim/quantizing/qat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THU-MIG/torch-model-compression/HEAD/torchslim/quantizing/qat.py -------------------------------------------------------------------------------- /torchslim/quantizing/qat_tools.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THU-MIG/torch-model-compression/HEAD/torchslim/quantizing/qat_tools.py -------------------------------------------------------------------------------- /torchslim/quantizing/quantizer_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THU-MIG/torch-model-compression/HEAD/torchslim/quantizing/quantizer_test.py -------------------------------------------------------------------------------- /torchslim/reparameter/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /torchslim/reparameter/reparam.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THU-MIG/torch-model-compression/HEAD/torchslim/reparameter/reparam.py -------------------------------------------------------------------------------- /torchslim/slim_solver.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THU-MIG/torch-model-compression/HEAD/torchslim/slim_solver.py --------------------------------------------------------------------------------