MSBench is an open-source model sparsification toolkit based on PyTorch fx.
git clone https://github.com/ModelTC/msbench.git
cd msbench/applications/imagenet_example/SAT
# modify run_uniform.sh
sh run_uniform.sh
This project is released under the Apache 2.0 license.