forked from ARM-software/ML-zoo
-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Merge pull request ARM-software#46 from ARM-software/transformer-tuto…
…rials Added transformers quantization guide
Showing
7 changed files
with
6,652 additions
and
0 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,10 @@ | ||
# Transformer Tutorials | ||
|
||
Jupiter notebooks showing how to quantise and compress toy transformer encoder and encoder-decoder models. | ||
|
||
# Tutorials | ||
|
||
* ViT_PCQAT.ipynb - Shows PCQAT using TFMOT for transformer encoder models. | ||
* ViT_2x4-PQAT.ipynb - Shows new 2x4 pruning and QAT with TFMOT for transformer encoder models. | ||
* translation.ipynb - Shows how to do QAT with TFMOT for encoder-decoder models. | ||
* translation_PQAT.ipynb - Shows how to apply pruning & QAT with TFMOT for encoder-decoder models. |
Oops, something went wrong.