weixin_42393848的博客

文章目录介绍实验代码结果介绍self.modules()是继承torch.nn.Modules()的类拥有的方法，以迭代器形式返回此前声明的所有layers实验代码class BasicBlock(nn.Module): expansion = 1 def __init__(self, inplanes, planes, stride=1, downsample=None): super(BasicBlock, self).__init__() se

文章目录作用使用方法作用可以添加一系列的模型序列（卷积层，全连接层等），input被Sequential的forward函数接收后，会传给序列的第一个模型，然后一个接一个，输入接输出，最后会输出Sequential的最后一个模型的输出结果使用方法# AlexNet的网络结构net = nn.Sequential( nn.Conv2d(3, 64, kernel_size=11, stride=4, padding=2), nn.ReLU(inplace=True),