Downloads · 30 days
111
2% of all-time downloads
resnet18_cifar10
SamAdamDay/resnet18_cifar10
resnet18_cifar10 is a image classification model from SamAdamDay. Use it when you need a label for an image. It is set up for timm. The card lists the license as mit.
This is a resnet18 model trained on the cifar10 dataset. To load this model use the timm library and run the following code:
All-time downloads
4.8K
Public
Parameters
11.2M
273 MB on disk
Likes
0
Public
Repo makeup
Click a slice to open those files.
.bin44.8 MB · 50%
At a glance
- Task
- Image Classification
- Library
- timm
- License
- mit
- Access
- Public
- Created
- Jun 25, 2024
- Updated
- Jul 1, 2024
- SHA
- 0c181237
Datasets
Model card
From the Hugging Face model README
Model card for resnet18_cifar10
This is a resnet18 model trained on the cifar10 dataset.
To load this model use the timm library and run the following code:
import timm
model = timm.create_model("hf_hub:SamAdamDay/resnet18_cifar10", pretrained=True)
The model was trained using the following command:
./distributed_train.sh --dataset torch/cifar10 --data-dir /root/data --dataset-download --model resnet18 --lr-base 0.3 --epochs 100 --input-size 3 256 256 -mean 0.49139968 0.48215827 0.44653124 --std 0.24703233 0.24348505 0.26158768 --num-classes 10
Metrics
The model has a test accuracy of 94.73.
Model Details
- Dataset: cifar10
- Number of epochs: 100
- Batch size: 128
- Base LR: 0.3
- LR scheduler: cosine
- Input size (3, 256, 256), images are scaled to this size
- PyTorch version: 2.3.0+cu121
- timm version: 1.0.7
- Task
- Image Classification
- Library
- timm
- License
- mit
- Created
- Jun 25, 2024
- Updated
- Jul 1, 2024