Sonal Kumar
first commit
ed7a497
|
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.