每天五分钟深度学习框架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是不要列表只要列表中的元素
代码分析：
我们先构建一个卷积层