VGG16 加BN层 可以基于这个去改

#定义架构，这里选用VGG16
class Vgg16(nn.Module):
    def __init__(self):
        super().__init__()
        self.conv1 = nn.Conv2d(3,64,3,padding=1)
        self.bach1 =  nn.BatchNorm2d(64) 
        self.conv2 = nn.Conv2d(64,64,3,padding=1)
        self.bach2 =  nn.BatchNorm2d(64) 
        self.pool1 = nn.MaxPool2d(2)

        self.conv3 = nn.Conv2d(64,128,3,padding=1)
        self.bach3 =  nn.BatchNorm2d(128) 
        self.conv4 = nn.Conv2d(128,128,3,padding=1)
        self.bach4 =  nn.BatchNorm2d(128) 
        self.pool2 = nn.MaxPool2d(2)