Downloads · 30 days
1.9K
8% of all-time downloads
edadaltocg/resnet18_cifar10
resnet18_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 resnet18 trained on cifar10.
Downloads · 30 days
1.9K
8% of all-time downloads
All-time downloads
23.1K
Public
Repo size
89.5 MB
Likes
10
Trending 1
Click a slice to open those files.
.bin44.8 MB · 100%
From the Hugging Face model README
This model is a small resnet18 trained on cifar10.
Use the code below to get started with the model.
import detectors
import timm
model = timm.create_model("resnet18_cifar10", pretrained=True)
Training data is cifar10.
config: scripts/train_configs/cifar10.json
model: resnet18_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 is cifar10.
This model card was created by Eduardo Dadalto.