MSBench is an open-source model sparsification toolkit based on PyTorch fx. MSBench is consists of SATBench and PTSBench. Recently, we have been gradually improving the code for PTSBench.

git clone https://github.com/ModelTC/msbench.git
cd msbench/applications/imagenet_example/PTS
# modify run_uniform.sh
sh run_uniform.sh

This project is released under the Apache 2.0 license.