[人工智能]深度学习框架 Pytorch（LeNet）

Pytorch Tensor 的通道排序：[batch, channel, height, width]
eg. 输入100张32*32的彩色图片
通道为[100，3，32，32]

搭建流程：
定义一个class

class LeNet(nn.Module):

定义初始化函数

 def __init__(self):
    super(LeNet, self).__init__()

定义正向传播的过程

  def forward(self, x):

LeNet网络：

下载训练集：
第一次使用时要将download设置为True才会自动去下载数据集

train_set = torchvision.datasets.CIFAR10(root='./data', train=True,
                                         download=True, transform=transform)

transform函数对图像进行预处理

 transform = transforms.Compose(
    [transforms.ToTensor(),
     transforms.Normalize((0.5, 0.5, 0.5), (0.5, 0.5, 0.5))])