Skip to content

Commit

Permalink
An example for Huggingface Bert
Browse files Browse the repository at this point in the history
  • Loading branch information
VainF authored Aug 14, 2023
1 parent 2cc7dcf commit f662b95
Showing 1 changed file with 6 additions and 1 deletion.
7 changes: 6 additions & 1 deletion examples/hf_transformers/readme.md
Original file line number Diff line number Diff line change
@@ -1,6 +1,11 @@
# Example for HuggingFace ViT

## Pruning
## Pruning HF ViT
```bash
python prune_vit.py
```

## Pruning HF Bert
```bash
python prune_bert.py
```

0 comments on commit f662b95

Please sign in to comment.