13. 神经网络基本骨架--nn.Module
nn.Module的基本使用
1. 什么是nn.Module
-
nn.Module模块式神经网络模型的基类，它提供了一些很简单方便的结构，来构建一个神经网络 -
在pytorch的官方文档(https://pytorch.org/docs/stable/generated/torch.nn.Module.html#torch.nn.Module)中提供了一个基于
nn.Module的简单样例，这也将是一个基本的前向传播结构import torch.nn as nn import torch.nn.functional as F class Model(nn.Module): def __init__(self): super().__init__() self.conv1 = nn.Conv2d(1, 20, 5) self.conv2 = nn.Conv2d(20, 20, 5) def forward(self, x): x = F.relu(self.conv1(x)) return F.relu(self.conv2(x)