├── LICENSE ├── README.md ├── classification └── pruner.py ├── examples ├── cifar100_prune.py ├── cifar100_resnet.py ├── cifar_resnet.py └── docs │ ├── concat.png │ ├── conv-conv.png │ ├── conv-fc.png │ ├── dep1.png │ ├── dep2.png │ ├── dep3.png │ ├── residual.png │ └── split.png ├── prune_demo.py ├── quant_demo.py ├── requirements.txt ├── setup.py ├── torch_pruning ├── __init__.py ├── autoslim.py ├── autoslim_test.py ├── dependency.py ├── flops_counter.py ├── prune │ ├── __init__.py │ ├── __pycache__ │ │ ├── __init__.cpython-36.pyc │ │ ├── __init__.cpython-38.pyc │ │ ├── structured.cpython-36.pyc │ │ ├── structured.cpython-38.pyc │ │ ├── unstructured.cpython-36.pyc │ │ └── unstructured.cpython-38.pyc │ ├── structured.py │ └── unstructured.py ├── resnet_small.py ├── sensitivity_analysis.py └── utils.py └── torch_quanting ├── __init__.py ├── autoquant.py └── quantizer.py /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yanggui19891007/Pytorch-Auto-Slim-Tools/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yanggui19891007/Pytorch-Auto-Slim-Tools/HEAD/README.md -------------------------------------------------------------------------------- /classification/pruner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yanggui19891007/Pytorch-Auto-Slim-Tools/HEAD/classification/pruner.py -------------------------------------------------------------------------------- /examples/cifar100_prune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yanggui19891007/Pytorch-Auto-Slim-Tools/HEAD/examples/cifar100_prune.py -------------------------------------------------------------------------------- /examples/cifar100_resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yanggui19891007/Pytorch-Auto-Slim-Tools/HEAD/examples/cifar100_resnet.py -------------------------------------------------------------------------------- /examples/cifar_resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yanggui19891007/Pytorch-Auto-Slim-Tools/HEAD/examples/cifar_resnet.py -------------------------------------------------------------------------------- /examples/docs/concat.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yanggui19891007/Pytorch-Auto-Slim-Tools/HEAD/examples/docs/concat.png -------------------------------------------------------------------------------- /examples/docs/conv-conv.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yanggui19891007/Pytorch-Auto-Slim-Tools/HEAD/examples/docs/conv-conv.png -------------------------------------------------------------------------------- /examples/docs/conv-fc.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yanggui19891007/Pytorch-Auto-Slim-Tools/HEAD/examples/docs/conv-fc.png -------------------------------------------------------------------------------- /examples/docs/dep1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yanggui19891007/Pytorch-Auto-Slim-Tools/HEAD/examples/docs/dep1.png -------------------------------------------------------------------------------- /examples/docs/dep2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yanggui19891007/Pytorch-Auto-Slim-Tools/HEAD/examples/docs/dep2.png -------------------------------------------------------------------------------- /examples/docs/dep3.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yanggui19891007/Pytorch-Auto-Slim-Tools/HEAD/examples/docs/dep3.png -------------------------------------------------------------------------------- /examples/docs/residual.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yanggui19891007/Pytorch-Auto-Slim-Tools/HEAD/examples/docs/residual.png -------------------------------------------------------------------------------- /examples/docs/split.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yanggui19891007/Pytorch-Auto-Slim-Tools/HEAD/examples/docs/split.png -------------------------------------------------------------------------------- /prune_demo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yanggui19891007/Pytorch-Auto-Slim-Tools/HEAD/prune_demo.py -------------------------------------------------------------------------------- /quant_demo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yanggui19891007/Pytorch-Auto-Slim-Tools/HEAD/quant_demo.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- 1 | torch>=1.0 -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yanggui19891007/Pytorch-Auto-Slim-Tools/HEAD/setup.py -------------------------------------------------------------------------------- /torch_pruning/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yanggui19891007/Pytorch-Auto-Slim-Tools/HEAD/torch_pruning/__init__.py -------------------------------------------------------------------------------- /torch_pruning/autoslim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yanggui19891007/Pytorch-Auto-Slim-Tools/HEAD/torch_pruning/autoslim.py -------------------------------------------------------------------------------- /torch_pruning/autoslim_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yanggui19891007/Pytorch-Auto-Slim-Tools/HEAD/torch_pruning/autoslim_test.py -------------------------------------------------------------------------------- /torch_pruning/dependency.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yanggui19891007/Pytorch-Auto-Slim-Tools/HEAD/torch_pruning/dependency.py -------------------------------------------------------------------------------- /torch_pruning/flops_counter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yanggui19891007/Pytorch-Auto-Slim-Tools/HEAD/torch_pruning/flops_counter.py -------------------------------------------------------------------------------- /torch_pruning/prune/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yanggui19891007/Pytorch-Auto-Slim-Tools/HEAD/torch_pruning/prune/__init__.py -------------------------------------------------------------------------------- /torch_pruning/prune/__pycache__/__init__.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yanggui19891007/Pytorch-Auto-Slim-Tools/HEAD/torch_pruning/prune/__pycache__/__init__.cpython-36.pyc -------------------------------------------------------------------------------- /torch_pruning/prune/__pycache__/__init__.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yanggui19891007/Pytorch-Auto-Slim-Tools/HEAD/torch_pruning/prune/__pycache__/__init__.cpython-38.pyc -------------------------------------------------------------------------------- /torch_pruning/prune/__pycache__/structured.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yanggui19891007/Pytorch-Auto-Slim-Tools/HEAD/torch_pruning/prune/__pycache__/structured.cpython-36.pyc -------------------------------------------------------------------------------- /torch_pruning/prune/__pycache__/structured.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yanggui19891007/Pytorch-Auto-Slim-Tools/HEAD/torch_pruning/prune/__pycache__/structured.cpython-38.pyc -------------------------------------------------------------------------------- /torch_pruning/prune/__pycache__/unstructured.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yanggui19891007/Pytorch-Auto-Slim-Tools/HEAD/torch_pruning/prune/__pycache__/unstructured.cpython-36.pyc -------------------------------------------------------------------------------- /torch_pruning/prune/__pycache__/unstructured.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yanggui19891007/Pytorch-Auto-Slim-Tools/HEAD/torch_pruning/prune/__pycache__/unstructured.cpython-38.pyc -------------------------------------------------------------------------------- /torch_pruning/prune/structured.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yanggui19891007/Pytorch-Auto-Slim-Tools/HEAD/torch_pruning/prune/structured.py -------------------------------------------------------------------------------- /torch_pruning/prune/unstructured.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yanggui19891007/Pytorch-Auto-Slim-Tools/HEAD/torch_pruning/prune/unstructured.py -------------------------------------------------------------------------------- /torch_pruning/resnet_small.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yanggui19891007/Pytorch-Auto-Slim-Tools/HEAD/torch_pruning/resnet_small.py -------------------------------------------------------------------------------- /torch_pruning/sensitivity_analysis.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yanggui19891007/Pytorch-Auto-Slim-Tools/HEAD/torch_pruning/sensitivity_analysis.py -------------------------------------------------------------------------------- /torch_pruning/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yanggui19891007/Pytorch-Auto-Slim-Tools/HEAD/torch_pruning/utils.py -------------------------------------------------------------------------------- /torch_quanting/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yanggui19891007/Pytorch-Auto-Slim-Tools/HEAD/torch_quanting/__init__.py -------------------------------------------------------------------------------- /torch_quanting/autoquant.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yanggui19891007/Pytorch-Auto-Slim-Tools/HEAD/torch_quanting/autoquant.py -------------------------------------------------------------------------------- /torch_quanting/quantizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yanggui19891007/Pytorch-Auto-Slim-Tools/HEAD/torch_quanting/quantizer.py --------------------------------------------------------------------------------