基于 PyTorch 的 CIFAR-10 图像分类项目,使用卷积神经网络(CNN)对 10 个类别的图像进行分类。 ├── model.py # 模型定义 ├── train.py # CPU 训练 ├── train_gpu_2.py # GPU 训练(推荐) ├── dataset_transform.py # 数据集处理 ├── dataloader.py # 数据加载 ├── ...