Pytorch 实现全连接神经网络/卷积神经网络训练MNIST数据集,并将训练好的模型在制作自己的手写图片数据集上测试
使用教程代码下载地址:点我下载模型在训练过程中会自动显示训练进度,如果您的pytorch是CPU版本的,代码会自动选择CPU训练,如果有cuda,则会选择GPU训练。项目目录说明:CNN文件夹是用来保存卷积神经网络模型代码,其中model.py,my_dataset.py是被自动调用的,都不需要运行FC文件夹是用来保存全连接神经网络模型代码,其中model.py,my_dataset.py是被自动