使用PyTorch训练简单CIFAR10图片分类器
- main.py:与训练相关的文件(包含数据集加载、优化器、损失函数定义等)
- test.py:为测试文件 (数据集可以直接运行main.py文件进行加载,其他细节说明可以参考链接1)
根据以下两个教程进行第一个pytorch搭建(使用cifar10数据集): (1)如何使用PyTorch训练简单CIFAR10图片分类器(保姆级) 如何使用PyTorch训练简单CIFAR10图片分类器(保姆级)
(2)pytorch搭建神经网络:从入门到精通,细节 make your network more perfectpytorch搭建神经网络:从入门到精通,细节 make your network more perfect