nn.batchnorm2d

nn.Conv2d(in_channel, out_channel, 3, stride, 1, bias=False),
nn.BatchNorm2d(out_channel),         #BatchNorm2d最常用于卷积网络中(防止梯度消失或爆炸)，设置的参数就是卷积的输出通道数

nn.ReLU(inplace=True),

http://www.mamicode.com/info-detail-2378483.html     好文mark