Downloads · 30 days
0
Classifiers-Enhanced-by-Pre-training
RyukiRi/Classifiers-Enhanced-by-Pre-training
Classifiers-Enhanced-by-Pre-training is a machine learning model from RyukiRi. Use it for the machine learning task on the model card, and read the license before you ship it in a product. The card lists the license as mit.
This project utilizes a visual encoder from the pre-trained CLIP (ViT-B/32) to build image classifiers. To use the trained models, follow the steps below to set up and run the classifiers.
Access
Public
Updated Apr 16, 2024
Repo size
1.2 GB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.pth1.1 GB · 86%
At a glance
- License
- mit
- Access
- Public
- Created
- Apr 16, 2024
- Updated
- Apr 16, 2024
- SHA
- 429f7170
Model card
From the Hugging Face model README
Classifiers Enhanced by Pre-training
This project utilizes a visual encoder from the pre-trained CLIP (ViT-B/32) to build image classifiers. To use the trained models, follow the steps below to set up and run the classifiers.
Prerequisites
Before you start, make sure you have Python and the necessary libraries installed.
Download the Trained Models and CIFAR-100 Dataset
You need to download the following trained model weights and CIFAR-100 dataset for running the project:
fine-tune-best.pth: Best model weights after fine-tuning.linear-probe-best.pth: Best model weights after the linear probe training.train-from-scratch-best.pth: Best model weights trained from scratch.
Please download these files and place them under the results/ directory within the project folder.
cifar-100-python.tar.gz: CIFAR-100 dataset.
Please download this file and place it under the data/ directory within the project folder.
Installation and Usage
See https://github.com/Gengsheng-Li/Classifiers-enhanced-by-pre-training for more details.
- License
- mit
- Created
- Apr 16, 2024
- Updated
- Apr 16, 2024