├── .gitignore ├── LICENSE ├── README.md ├── proxssi ├── __init__.py ├── groups │ ├── __init__.py │ ├── bert.py │ ├── resnet.py │ └── vgg.py ├── l1_l2.py ├── mcp_l2.py ├── optimizers │ ├── __init__.py │ └── adamw_hf.py ├── penalties.py └── tests │ ├── __init__.py │ ├── conftest.py │ ├── test_bert.py │ ├── test_l1_l2.py │ ├── test_mcp_l2.py │ └── test_vgg.py └── requirements.txt /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tristandeleu/pytorch-structured-sparsity/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tristandeleu/pytorch-structured-sparsity/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tristandeleu/pytorch-structured-sparsity/HEAD/README.md -------------------------------------------------------------------------------- /proxssi/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /proxssi/groups/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tristandeleu/pytorch-structured-sparsity/HEAD/proxssi/groups/__init__.py -------------------------------------------------------------------------------- /proxssi/groups/bert.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tristandeleu/pytorch-structured-sparsity/HEAD/proxssi/groups/bert.py -------------------------------------------------------------------------------- /proxssi/groups/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tristandeleu/pytorch-structured-sparsity/HEAD/proxssi/groups/resnet.py -------------------------------------------------------------------------------- /proxssi/groups/vgg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tristandeleu/pytorch-structured-sparsity/HEAD/proxssi/groups/vgg.py -------------------------------------------------------------------------------- /proxssi/l1_l2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tristandeleu/pytorch-structured-sparsity/HEAD/proxssi/l1_l2.py -------------------------------------------------------------------------------- /proxssi/mcp_l2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tristandeleu/pytorch-structured-sparsity/HEAD/proxssi/mcp_l2.py -------------------------------------------------------------------------------- /proxssi/optimizers/__init__.py: -------------------------------------------------------------------------------- 1 | from proxssi.optimizers.adamw_hf import AdamW -------------------------------------------------------------------------------- /proxssi/optimizers/adamw_hf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tristandeleu/pytorch-structured-sparsity/HEAD/proxssi/optimizers/adamw_hf.py -------------------------------------------------------------------------------- /proxssi/penalties.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tristandeleu/pytorch-structured-sparsity/HEAD/proxssi/penalties.py -------------------------------------------------------------------------------- /proxssi/tests/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tristandeleu/pytorch-structured-sparsity/HEAD/proxssi/tests/__init__.py -------------------------------------------------------------------------------- /proxssi/tests/conftest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tristandeleu/pytorch-structured-sparsity/HEAD/proxssi/tests/conftest.py -------------------------------------------------------------------------------- /proxssi/tests/test_bert.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tristandeleu/pytorch-structured-sparsity/HEAD/proxssi/tests/test_bert.py -------------------------------------------------------------------------------- /proxssi/tests/test_l1_l2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tristandeleu/pytorch-structured-sparsity/HEAD/proxssi/tests/test_l1_l2.py -------------------------------------------------------------------------------- /proxssi/tests/test_mcp_l2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tristandeleu/pytorch-structured-sparsity/HEAD/proxssi/tests/test_mcp_l2.py -------------------------------------------------------------------------------- /proxssi/tests/test_vgg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tristandeleu/pytorch-structured-sparsity/HEAD/proxssi/tests/test_vgg.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tristandeleu/pytorch-structured-sparsity/HEAD/requirements.txt --------------------------------------------------------------------------------