Hugging Face's logo

ashhadahsan
/
sam_vit_b_lora_rank512_ufpr

Safetensors
English
German
Model card Files Community
sam_vit_b_lora_rank512_ufpr
File size: 820 Bytes 
be91aef
 
31a1c50
 
be91aef
31a1c50
be91aef
31a1c50
713c2da
 
 
 
 
 
 
 
 
 
 
 
65013fd
7f6ec43
65013fd
7f6ec43
 
 
 
 
 
 
 
 
 
 
 
 
61ef435
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
 
---
language:
  - en
  - de
base_model:
  - facebook/sam-vit-base
---

![Training Loss UFPR SAM](https://huggingface.co/ashhadahsan/sam_vit_b_lora_rank512_ufpr/resolve/main/training_loss_ufpr_sam.png)

![Comparison with baseline](https://huggingface.co/ashhadahsan/sam_vit_b_lora_rank512_ufpr/resolve/main/comparsion_with_baseline.png)

## Finetuning metrics

* RANK LOSS : 0.052654343576998344			
* RANK ACCURACY : 0.9997852160627786			
* RANK PRECISION : 0.9420321321139293		
* RANK RECALL : 0.9461407410472382			


### Parameters on which the model was trained

```yaml
DATASET:
  TRAIN_PATH: "./ufpr_dataset/training"
  TEST_PATH: "./ufpr_dataset/testing"
  IMAGE_FORMAT: ".png"

SAM:
  CHECKPOINT: "./sam_vit_b_01ec64.pth"
  RANK: 512

TRAIN:
  BATCH_SIZE: 1
  NUM_EPOCHS: 30
  LEARNING_RATE: 0.0001
  PATIENCE: 10