每天五分钟深度学习框架pytorch:快速搭建VGG网络的基础模块VGG块

本文重点

前面我们介绍了VGG神经网络，我们知道VGG是由许多的VGG块构成，那么本文我们将使用pytorch搭建VGG块

代码实现：

import torch

from torch import nn

def vgg_block(num_convs,in_channels,out_channels):

net=[nn.Conv2d(in_channels,out_channels,kernel_size=3,padding=1),nn.ReLU(inplace=True)]

for i in range(num_convs-1):

net.append(nn.Conv2d(out_channels,out_channels,kernel_size=3,padding=1)) net.append(nn.ReLU(inplace=True)) net.append(nn.MaxPool2d(2,2))

return nn.Sequential(*net)#*net是不要列表只要列表中的元素

代码分析：

我们先构建一个卷积层