Downloads · 30 days
0
cifar10-efficientnet-b0-transfer-learning
aijadugar/cifar10-efficientnet-b0-transfer-learning
cifar10-efficientnet-b0-transfer-learning is a machine learning model from aijadugar. Use it for the machine learning task on the model card, and read the license before you ship it in a product.
This project demonstrates Transfer Learning using a pretrained EfficientNet-B0 model on the CIFAR-10 dataset with PyTorch.
Access
Public
Updated Jul 21, 2026
Repo size
33.2 MB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.pth32.8 MB · 99%
At a glance
- Access
- Public
- Created
- Jul 21, 2026
- Updated
- Jul 21, 2026
- SHA
- 9e7f8ca9
Model card
From the Hugging Face model README
Transfer Learning with EfficientNet-B0 on CIFAR-10
This project demonstrates Transfer Learning using a pretrained EfficientNet-B0 model on the CIFAR-10 dataset with PyTorch.
Two transfer learning approaches are implemented:
- Feature Extraction
- Fine-Tuning
Dataset
- Dataset: CIFAR-10
- Training Images: 45,000
- Validation Images: 5,000
- Test Images: 10,000
- Classes: 10
Training Pipeline
Feature Extraction
- Load pretrained EfficientNet-B0
- Replace the final classifier
- Freeze the backbone
- Train only the classifier
Fine-Tuning
- Load the best feature extraction model
- Unfreeze features[8] and the classifier
- Fine-tune using different learning rates
Results
Feature Extraction
| Loss | Accuracy |
|---|---|
 |  |
Fine-Tuning
| Loss | Accuracy |
|---|---|
 |  |
Technologies
- Python
- PyTorch
- TorchVision
- NumPy
- Matplotlib
- tqdm
Author
Ankit Bari
- Created
- Jul 21, 2026
- Updated
- Jul 21, 2026