Downloads · 30 days
39
4% of all-time downloads
edadaltocg/densenet121_svhn
densenet121_svhn 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 svhn.
Downloads · 30 days
39
4% of all-time downloads
All-time downloads
1.1K
Public
Repo size
8.6 MB
Likes
0
Public
Click a slice to open those files.
.bin4.4 MB · 100%
From the Hugging Face model README
This model is a small densenet121 trained on svhn.
Use the code below to get started with the model.
import detectors
import timm
model = timm.create_model("densenet121_svhn", pretrained=True)
Training data is svhn.
config: scripts/train_configs/svhn.json
model: densenet121_svhn
dataset: svhn
batch_size: 128
epochs: 300
validation_frequency: 5
seed: 1
criterion: CrossEntropyLoss
criterion_kwargs: {}
optimizer: SGD
lr: 0.01
optimizer_kwargs: {'momentum': 0.9, 'weight_decay': 0.0005}
scheduler: MultiStepLR
scheduler_kwargs: {'gamma': 0.1, 'milestones': [75, 100, 150, 225]}
debug: False
Testing data is svhn.
This model card was created by Eduardo Dadalto.