Skip to content

Commit

Permalink
update readme
Browse files Browse the repository at this point in the history
  • Loading branch information
VainF committed Jan 3, 2023
1 parent 15ce66b commit c31976f
Showing 1 changed file with 4 additions and 1 deletion.
5 changes: 4 additions & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -26,7 +26,10 @@ Torch-Pruning is a general-purpose library for structural network pruning, which
Torch-Pruning will forward your model with a fake inputs and trace the computational graph just like ``torch.jit``. A dependency graph will be established to record the relation coupling between layers. Torch-pruning will collect all affected layers according by propogating your pruning operations through the whole graph, and then return a `PruningClique` for pruning. All pruning indices will be automatically transformed if there are operations like ``torch.split`` or ``torch.cat``.

## Installation

```bash
pip install torch-pruning # v1.0.0
```
or
```bash
git clone https://github.com/VainF/Torch-Pruning.git
```
Expand Down

0 comments on commit c31976f

Please sign in to comment.