#17 · 主分类: 模型动物园与中心
PyTorch_CIFAR10
在CIFAR10数据集上预训练的TorchVision模型(带权重)
项目最后更新:06/24/23
GitHub Stars
696
Forks数量
161
贡献者数量
4
许可证
MIT
收录理由
做CIFAR-10相关实验时,如果不想从零训练分类器,这个仓库能省下不少时间。作者把TorchVision官方的CNN实现改造成适配10类CIFAR-10任务,并附上训练好的权重,加载后即可直接做推理。每个模型都列出了验证准确率、参数量和文件大小,方便你在精度和内存占用之间做权衡。训练代码基于PyTorch-Lightning,可复现性高,想自己重训或微调也能直接用同一套脚本。对于需要一个可靠CIFAR-10基线模型来继续开发的开发者,这是个很实用的起点。
相关文章
暂无针对该项目的文章。
如需补充新闻资讯、使用教程、应用测评等内容,欢迎 联系我们推荐选题或投稿。
同类热门项目
models.dev
一个开源的人工智能模型数据库。
models
由PaddlePaddle官方维护和支持,涵盖CV、NLP、语音、推荐、时间序列、大模型等。
Awesome-CoreML-Models
最全的 Core ML 模型列表(适用于 iOS 11+)
PINTO_model_zoo
一个用于存储已在各种框架之间相互转换的模型的仓库。支持的框架包括TensorFlow、PyTorch、ONNX、OpenVINO、TFJS、TFTRT、TensorFlowLite(Float32/16/INT8)、EdgeTPU、CoreML。
huggingface_hub
Hugging Face Hub 的官方 CLI 和 Python 客户端。