sam_vit_b_lora_rank512_ufpr

ashhadahsan/sam_vit_b_lora_rank512_ufpr

sam_vit_b_lora_rank512_ufpr is a machine learning model from ashhadahsan. Use it for the machine learning task on the model card, and read the license before you ship it in a product.

Downloads · 30 days

Access

Public

Updated Sep 28, 2024

Parameters

18.9M

75.5 MB on disk

Likes

Public

Hugging Face

Repo makeup

Click a slice to open those files.

.safetensors75.5 MB · 100%

At a glance

Access: Public
Created: Sep 28, 2024
Updated: Sep 28, 2024
SHA: 713c2da8

Base models

facebook/sam-vit-base

Model card

From the Hugging Face model README

Training Loss UFPR SAM

Comparison with baseline

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

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

Languages: en, de
Created: Sep 28, 2024
Updated: Sep 28, 2024