This is the official implementation of "Fast and Controllable Post-training Sparsity: Learning Optimal Sparsity Allocation with Global Constraint in Minutes"
The paper link is : FCPTS
The repo depends on msbench
git clone https://github.com/ModelTC/FCPTS.git
git clone https://github.com/ModelTC/msbench.git
cd FCPTS
# modify run.sh
bash run.sh
The core code:
class FCPTSMaskGenerator in msbench/mask_generator.py
class SparsityConstrain in FCPTS/main.py