Bootstrap

Pytorch使用手册-Build the Neural Network(专题五)

在 PyTorch 中如何构建一个用于 FashionMNIST 数据集分类的神经网络模型,并解析了 PyTorch 的核心模块 torch.nn 的使用方法。以下是具体内容的讲解:


构建神经网络

在 PyTorch 中,神经网络的核心在于 torch.nn 模块,它提供了构建神经网络所需的所有工具。关键点如下:

  • nn.Module: 所有神经网络的基类。
    • 神经网络是 nn.Module 的子类。
    • 它支持将多个模块(如层、激活函数)嵌套在一起,方便构建复杂网络。
  • nn.Sequential: 一个按顺序组织模块的容器,用于快速构建简单网络。

1. 设备选择

为了高效训练模型,可以选择硬件加速(如 GPU 或 Apple MPS),否则使用 CPU:

device = 
;