Skip to content
/ cahsi-project Public

Applying LoRA to efficiently fine-tune SAM on covid-19 chest x-rays

0 stars 0 forks Branches Tags Activity
Star
Notifications You must be signed in to change notification settings

oliverc1623/cahsi-project

BranchesTags

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

56 Commits
notebooks
notebooks
 
 
scripts
scripts
 
 
CAHSI Project Report.ipynb
CAHSI Project Report.ipynb
 
 
CAHSI Project Report.pdf
CAHSI Project Report.pdf
 
 
README.md
README.md
 
 
cahsi-report.md
cahsi-report.md
 
 
environment.yml
environment.yml
 
 
lora-qata-sam-finetune.py
lora-qata-sam-finetune.py
 
 
packages.txt
packages.txt
 
 
pod-environment.yml
pod-environment.yml
 
 
run2.md
run2.md
 
 
train_losses.png
train_losses.png
 
 

Repository files navigation

Cahsi Project

Applying LoRA to efficiently fine-tune SAM on covid-19 chest x-rays.

TODOs:

  • Apply LoRA to all MLP layers
  • Finetune Vision encoder, in addition to mask decoder
  • Apply LoRA to LayerNorm layers
  • Fine-tune on entire dataset
  • Add validation set to check overfitting

About

Applying LoRA to efficiently fine-tune SAM on covid-19 chest x-rays

Topics

pytorch lora image-segmentation covid-19 segment-anything

Resources

Readme
Activity

Stars

0 stars

Watchers

1 watching

Forks

0 forks
Report repository

Releases

No releases published

Packages

No packages published

Languages