【深度之眼】Pytorch框架班第五期-模型容器之nn.Sequential代码调试

模型容器(Containers)

nn.Sequential

nn.Sequential 是 nn.module 的容器，用于按顺序包装一组网络层

• 顺序性：各网络层之间严格按照顺序构建
• 自带forward(): 自带的forward里，通过for循环一次执行前向传播运算。

class LeNetSequential(nn.Module):
    def __init__(self, classes):
        super(LeNetSequential, self).__init__()
        self.features = nn.Sequential(
            nn.Conv2d(3, 6, 5),
            nn.ReLU(),
            nn.MaxPool2d(kernel_size=2, stride=2),
            nn.Conv2d(6, 16, 5),
            nn.ReLU(),
            nn.MaxPool2d(kernel_size=2, stride=2),)

        self