pytorch实现MNIST手写体识别(全连接神经网络)
环境: pytorch1.1 cuda9.0 ubuntu16.04 该网络有3层,第一层input layer,有784个神经元(MNIST数据集是28*28的单通道图片,故有784个神经元)。第二层为hidden_layer,设置为500个神经元。最后一层是输出层,有10个神经元(10分类任务)。在第二层之后还有个ReLU函数,进行非线性变换。 #!/usr/bin/env python # encoding: utf-8 ''' @author: liualex @contact: liualex1109@163.com @software: pycharm @file: main.py @time: 2019/8/14 21:20 @desc: ''' import torch import torchvision import torchvision.transforms as transforms import torch.utils.data.dataloader as dataloader import torch.nn as nn import torch.optim as optim import os os.environ["CUDA_VISIBLE_DEVICES"] = "3" train_set = torchvision.datasets