Spaces:
sonalkum
/
Running on Zero

sonalkum's picture
hf peft
1e6d67a
|
raw
history blame
888 Bytes

Fine-tuning for semantic segmentation using LoRA and 🤗 PEFT

Open In Colab

We provide a notebook (semantic_segmentation_peft_lora.ipynb) where we learn how to use LoRA from 🤗 PEFT to fine-tune an semantic segmentation by ONLY using 14%% 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.