深度学习工具Pytorch—用于实现LeNet-5

Pytorch框架实现LeNet-5

• 1. 背景介绍及目标设定

  • 2. 基于LeNet框架的PyTorch技术实现

  • • 2.1 主程序模块main()

  • 2.2 分别负责训练过程与验证流程的训练模块train()及测试模块test()

  • 3. 结尾

  • 参考资料

1.背景与目标

在本节课程中，我们将深入探讨当下备受关注的深度学习框架Pytorch的实际应用方法。近年来Pytorch的使用份额呈现出显著的增长态势，在学术界已逐渐被公认为深度学习领域的重要研究平台。本次课程中将通过LeNet网络为例演示Pytorch的具体操作流程。

2. 基于LeNet的Pytorch实现

2.1 主函数main()

首先，我们打开main.py程序：

    import torch
    from torch.utils.data import DataLoader
    import scipy.misc
    from torch.utils.data import DataLoader
    from torchvision import transforms
    from torchvision.datasets import MNIST
    fr