从LeNet-5 CNN模型入门PyTorch
1. PyTorch 准备
1.1 PyTorch特点
1.2 PyTorch安装测试
2. 完整代码
2.1 LeNet模型
2.2 训练
2.2 测试
3. 解读PyTorch
3.1 CNN Api
3.1.1 二维卷积Conv2d
3.1.2 激活函数ReLU
3.1.3 池化层函数MaxPool2d
3.1.4 全连接层Linear
3.1.5 Module父类
3.2 Basic Api
3.2.1 改变Tensor shape的view
3.2.2 torch.optim.SGD
3.2.3 torch.nn.CrossEntropyLoss
3.2.4 保存模型torch.save
3.2.5 torch.Tensor.detach
从LeNet-5 CNN模型入门PyTorch
1. PyTorch 准备
1.1 PyTorch
阅读背景：
从LeNet-5 CNN模型入门PyTorch
来源：互联网
分享到：
