Downloads · 30 days
454
6% of all-time downloads
edadaltocg/resnet50_cifar10
resnet50_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 resnet50 trained on cifar10.
Downloads · 30 days
454
6% of all-time downloads
All-time downloads
7.3K
Public
Repo size
189 MB
Likes
1
Public
Click a slice to open those files.
.bin94.4 MB · 100%
From the Hugging Face model README
This model is a small resnet50 trained on cifar10.
Use the code below to get started with the model.
import detectors
import timm
model = timm.create_model("resnet50_cifar10", pretrained=True)
Training data is cifar10.
config: scripts/train_configs/cifar10.json
model: resnet50_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.