Downloads · 30 days
0
bird_classifier_lora
mibo222/bird_classifier_lora
bird_classifier_lora is a image classification model from mibo222. Use it when you need a label for an image. It is set up for peft. The card lists the license as mit.
This repository contains LoRA (Low-Rank Adaptation) adapters fine-tuned on the CUB-200-2011 dataset for bird image classification. These adapters are designed to be applied to a standard torchvision.models.resnet50 ba…
Access
Public
Updated May 5, 2025
Repo size
96.6 MB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.pth96 MB · 99%
At a glance
- Task
- Image Classification
- Library
- peft
- License
- mit
- Access
- Public
- Created
- May 5, 2025
- Updated
- May 5, 2025
- SHA
- d6d2bfbc
Datasets
Model card
From the Hugging Face model README
ResNet50 + LoRA for Bird Classification (CUB-200-2011)
This repository contains LoRA (Low-Rank Adaptation) adapters fine-tuned on the CUB-200-2011 dataset for bird image classification. These adapters are designed to be applied to a standard torchvision.models.resnet50 base model.
Model Details
- Base Model:
torchvision.models.resnet50(pre-trained on ImageNet). - Fine-tuning Method: Low-Rank Adaptation (LoRA) using the
peftlibrary. - Dataset: Caltech-UCSD Birds-200-2011 (CUB-200-2011)
- Number of Classes: 200 bird species.
- LoRA Configuration:
- Rank (
r): 8 (as used in training, please verify/update) - Alpha (
lora_alpha): 16 (as used in training, please verify/update) - Target Modules: ["fc", "conv1", "layer4.0.conv1"] (Please list the actual modules targeted during training)
- Dropout: 0.05
- Bias: "none"
- Rank (
How to Use
First, make sure you have torch, torchvision, and peft installed:
pip install torch torchvision peft Pillow
- Task
- Image Classification
- Library
- peft
- License
- mit
- Languages
- en
- Created
- May 5, 2025
- Updated
- May 5, 2025