Downloads · 30 days
1
5% of all-time downloads
vit-base-oxford-iiit-pets
schuegia/vit-base-oxford-iiit-pets
vit-base-oxford-iiit-pets is a machine learning model from schuegia. Use it for the machine learning task on the model card, and read the license before you ship it in a product.
1. Create a Space on Lightning AI. 2. Upload Files: - requirements.txt - transferlearningimageclassification.ipynb 3. Install Packages: Run the following command to install the required dependencies:
All-time downloads
19
Public
Parameters
85.8M
1.7 GB on disk
Likes
0
Public
Repo makeup
Click a slice to open those files.
.safetensors343 MB · 100%
At a glance
- Model type
- vit
- Access
- Public
- Created
- Apr 17, 2025
- Updated
- Apr 17, 2025
- SHA
- 32733e99
Model card
From the Hugging Face model README
Run using Lightning AI
Steps to Run:
- Create a Space on Lightning AI.
- Upload Files:
- Install Packages:
Run the following command to install the required dependencies:pip install -r requirements.txt - Run the Notebook.
Notes:
- The second code block in the notebook handles Hugging Face login, which is required to upload the trained model. You need to generate an access token from Hugging Face.
- It is highly recommended to use a GPU instead of a CPU, as training time is reduced by a factor of 24x when using a GPU.
model: ViTForImageClassification Accuracy: 0.8800 Precision: 0.8768 Recall: 0.8800
- Type
- vit
- Created
- Apr 17, 2025
- Updated
- Apr 17, 2025