middle_model.py 文件源码

def __init__(self, n_classes):
        super(MiddleCNN, self).__init__()
        with self.init_scope():
            self.conv1 = L.ConvolutionND(2, 3, 16, 3, pad=1, initialW=init())
            self.bnorm1 = L.BatchNormalization(16)
            self.conv2 = L.ConvolutionND(2, 16, 16, 3, pad=1, initialW=init())
            self.bnorm2 = L.BatchNormalization(16)
            self.conv3 = L.ConvolutionND(2, 16, 32, 3, pad=1, initialW=init())
            self.bnorm3 = L.BatchNormalization(32)
            self.conv4 = L.ConvolutionND(2, 32, 32, 3, pad=1, initialW=init())
            self.bnorm4 = L.BatchNormalization(32)
            self.fc = L.Linear(None, n_classes)