Home

Awesome

<div align="center"> <img src="resources/logo.png" width="600"/> </div>

Introduction

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.

Quick start pts example without installation

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

License

This project is released under the Apache 2.0 license.