pytorch-cifar100项目简介
pytorch-cifar100是一个基于PyTorch实现CIFAR-100数据集图像分类的开源项目。该项目实现了多种经典的深度学习网络模型,如ResNet、DenseNet、VGG等,为开发者提供了在CIFAR-100数据集上进行图像分类实验的代码范例。

项目主要特性
- 支持多种经典网络模型,包括ResNet、DenseNet、VGG、GoogleNet等
- 提供完整的训练和测试流程
- 可视化训练过程(使用TensorBoard)
- 支持GPU加速训练
使用指南
环境要求
- Python 3.6+
- PyTorch 1.6.0+
- TensorBoard 2.2.2 (可选)
快速开始
- 克隆项目仓库:
git clone https://github.com/weiaicunzai/pytorch-cifar100.git
cd pytorch-cifar100
- 安装依赖:
pip install -r requirements.txt
- 训练模型:
python train.py -net resne
8536





被折叠的 条评论
为什么被折叠?



