#17 · Primary category: Model Zoos & Hubs
PyTorch_CIFAR10
Pretrained TorchVision models on CIFAR10 dataset (with weights)
Project last updated:06/24/23
GitHub Stars
696
Forks
161
Contributors
4
License
MIT
Why we included this project
Anyone doing CIFAR-10 work who wants a pretrained classifier without training one from scratch will find this repo useful. It takes the official TorchVision CNN implementations, adjusts them for the 10-class CIFAR-10 task, and ships the resulting weights, so you can load a model and run inference right away. The README lists validation accuracy, parameter count, and file size for each network, which helps when you need to trade accuracy against memory. Training code runs on PyTorch-Lightning and is reproducible, so the same scripts work if you prefer to retrain or fine-tune. It is a handy baseline for people who want a known-good CIFAR-10 model to build on.
Articles for this project
No articles for this project yet.
To suggest a topic or contribute an article, contact us.
Related projects in this category
models.dev
An open-source database of AI models.
models
Officially maintained, supported by PaddlePaddle, including CV, NLP, Speech, Rec, TS, big models and so on.
Awesome-CoreML-Models
Largest list of models for Core ML (for iOS 11+)
PINTO_model_zoo
A repository for storing models that have been inter-converted between various frameworks. Supported frameworks are TensorFlow, PyTorch, ONNX, OpenVINO, TFJS, TFTRT, TensorFlowLite (Float32/16/INT8), EdgeTPU, CoreML.
huggingface_hub
The official CLI and Python client for the Hugging Face Hub.