VGG网络模型的搭建
class VGG(nn.Module):
def __init__(self):
super(VGG,self).__init__()
self.features=nn.Sequential( nn.Conv2d(1,64,kernel_size=3,padding=1), nn.ReLU(inplace=True), nn.Conv2d(64,64,kernel_size=3,padding=1), nn.ReLU(inplace=True), nn.MaxPool2d(kernel_size=2,stride=2), nn.Conv2d(64,128,kernel_size=3,padding=1), nn.ReLU(inplace=True), nn.Conv2d(128,128,kernel_size=3,padding=1), nn.ReLU(inplace=True), nn.MaxPool2d(kernel_size=2,stride=2), nn.Conv2d(128,256,kernel_size=3,padding=1), nn.ReLU(inplace=True), nn.Conv2d(256,256,kernel_size=3,padding=1), nn.ReLU(inplace=True), nn.Conv2d(256,256,kernel_size=4,padding