Merge branch 'main' into smangrul/fix-examples-with-hub-utils

This commit is contained in:
Sourab Mangrulkar
2023-02-08 18:41:19 +05:30
2 changed files with 14962 additions and 0 deletions
+7
View File
@@ -0,0 +1,7 @@
# Fine-tuning for image classification using LoRA and 🤗 PEFT
[![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/huggingface/peft/blob/main/examples/image_classification/image_classification_peft_lora.ipynb)
We provide a notebook (`image_classification_peft_lora.ipynb`) where we learn how to use [LoRA](https://arxiv.org/abs/2106.09685) from 🤗 PEFT to fine-tune an image classification modelby ONLY using **0.7%** of the original trainable parameters of the model.
LoRA adds low-rank "update matrices" to certain blocks in the underlying model (in this case the attention blocks) and ONLY trains those matrices during fine-tuning. During inference, these update matrices are _merged_ with the original model parameters. For more details, check out the [original LoRA paper](https://arxiv.org/abs/2106.09685).
File diff suppressed because one or more lines are too long