Files
peft/examples/image_classification/README.md
T
2023-02-07 09:55:07 +05:30

926 B

Fine-tuning for image classification using LoRA and 🤗 PEFT

Open In Colab

We provide a notebook (image_classification_peft_lora.ipynb) where we learn how to use LoRA from 🤗 PEFT to fine-tune an image classification modelby ONLY using 0.6% 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.