本文共 438 字，大约阅读时间需要 1 分钟。

2019.11.31

from d2l import torch as d2limport torchfrom torch import nndef vgg_block(num_convs, in_channels, out_channels):    layers = []    for _ in range(num_convs):        layers.append(nn.Conv2d(in_channels, out_channels,                                kernel_size=3, padding=1))        layers.append(nn.ReLU())        in_channels = out_channels    layers.append(nn.MaxPool2d(kernel_size=2,stride=2))    return nn.Sequential(*layers)

转载地址：http://erzai.baihongyu.com/