pytorch_cifar10

使用PyTorch训练简单CIFAR10图片分类器

使用说明

1. main.py:与训练相关的文件（包含数据集加载、优化器、损失函数定义等）
2. test.py:为测试文件 （数据集可以直接运行main.py文件进行加载，其他细节说明可以参考链接1）

运行效果图

训练集可视化：

测试结果：

参考链接

根据以下两个教程进行第一个pytorch搭建(使用cifar10数据集): (1)如何使用PyTorch训练简单CIFAR10图片分类器(保姆级) 如何使用PyTorch训练简单CIFAR10图片分类器(保姆级)

(2)pytorch搭建神经网络：从入门到精通，细节 make your network more perfectpytorch搭建神经网络：从入门到精通，细节 make your network more perfect