Downloads · 30 days
160
7% of all-time downloads
densenet121_cifar10
edadaltocg/densenet121_cifar10
densenet121_cifar10 is a image classification model from edadaltocg. Use it when you need a label for an image. It is set up for timm. The card lists the license as mit.
This model is a small densenet121 trained on cifar10.
All-time downloads
2.3K
Public
Repo size
8.6 MB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.bin4.4 MB · 100%
At a glance
- Task
- Image Classification
- Library
- timm
- License
- mit
- Access
- Public
- Created
- Feb 19, 2023
- Updated
- Feb 19, 2023
- SHA
- 3a729bcb
Datasets
Model card
From the Hugging Face model README
Model Card for Model ID
This model is a small densenet121 trained on cifar10.
- Test Accuracy: 0.9391
- License: MIT
How to Get Started with the Model
Use the code below to get started with the model.
import detectors
import timm
model = timm.create_model("densenet121_cifar10", pretrained=True)
Training Data
Training data is cifar10.
Training Hyperparameters
-
config:
scripts/train_configs/cifar10.json -
model:
densenet121_cifar10 -
dataset:
cifar10 -
batch_size:
128 -
epochs:
300 -
validation_frequency:
5 -
seed:
1 -
criterion:
CrossEntropyLoss -
criterion_kwargs:
{} -
optimizer:
SGD -
lr:
0.1 -
optimizer_kwargs:
{'momentum': 0.9, 'weight_decay': 0.0005, 'nesterov': 'True'} -
scheduler:
ReduceLROnPlateau -
scheduler_kwargs:
{'factor': 0.1, 'patience': 3, 'threshold': 0.001, 'mode': 'max'} -
debug:
False
Testing Data
Testing data is cifar10.
This model card was created by Eduardo Dadalto.
- Task
- Image Classification
- Library
- timm
- License
- mit
- Languages
- en
- Created
- Feb 19, 2023
- Updated
- Feb 19, 2023