#17 · 主分类: 模型动物园与中心

PyTorch_CIFAR10

cifar10 computer-vision deeplearning pretrained-models pretrained-weights pytorch torchvision

在CIFAR10数据集上预训练的TorchVision模型(带权重)

项目最后更新:06/24/23

GitHub Stars

696

Forks数量

161

贡献者数量

4

许可证

MIT

收录理由

做CIFAR-10相关实验时,如果不想从零训练分类器,这个仓库能省下不少时间。作者把TorchVision官方的CNN实现改造成适配10类CIFAR-10任务,并附上训练好的权重,加载后即可直接做推理。每个模型都列出了验证准确率、参数量和文件大小,方便你在精度和内存占用之间做权衡。训练代码基于PyTorch-Lightning,可复现性高,想自己重训或微调也能直接用同一套脚本。对于需要一个可靠CIFAR-10基线模型来继续开发的开发者,这是个很实用的起点。

相关文章

暂无针对该项目的文章。

如需补充新闻资讯、使用教程、应用测评等内容,欢迎 联系我们推荐选题或投稿。

同类热门项目