Downloads · 30 days
576
2% of all-time downloads
clip-vit-base-patch32_stl10
tanganke/clip-vit-base-patch32_stl10
clip-vit-base-patch32_stl10 is a feature extraction model from tanganke. Use it when you need embeddings to search or compare text. It is set up for transformers.
- Architecture: ViT-Base with patch size 32 - Training Data: STL10
All-time downloads
23.5K
Public
Parameters
87.5M
350 MB on disk
Likes
0
Public
Repo makeup
Click a slice to open those files.
.safetensors350 MB · 100%
At a glance
- Task
- Feature Extraction
- Library
- transformers
- Model type
- clip_vision_model
- Access
- Public
- Created
- Apr 29, 2024
- Updated
- Apr 29, 2024
- SHA
- 108786bc
Base models
Datasets
Model card
From the Hugging Face model README
Model Card
Model Details
- Architecture: ViT-Base with patch size 32
- Training Data: STL10
Training Details
Adam Optimizer with a constant learning rate 1e-5 for 4000 steps training (batch_size=32). Only the vision encoder is fine-tuned.
Evaluation Results
- pre-trained: 0.971250057220459
- fine-tuned: 0.9754999876022339
Usage
load vision model
from transformers import CLIPVisionModel
vision_model = CLIPVisionModel.from_pretrained('tanganke/clip-vit-base-patch32_stl10')
substitute the vision encoder of clip
from transformers import CLIPModel
clip_model = CLIPModel.from_pretrained("openai/clip-vit-base-patch32")
clip_model.vision_model.load_state_dict(vision_model.vision_model.state_dict())
- Task
- Feature Extraction
- Library
- transformers
- Type
- clip_vision_model
- Created
- Apr 29, 2024
- Updated
- Apr 29, 2024