simple_model.py 文件源码

def __init__(self, n_classes):
        super(SimpleCNN, self).__init__()
        with self.init_scope():
            self.conv1 = L.ConvolutionND(2, 3, 32, 3, pad=1, initialW=init())
            self.bnorm1 = L.BatchNormalization(32)
            self.conv2 = L.ConvolutionND(2, 32, 64, 3, pad=1, initialW=init())
            self.bnorm2 = L.BatchNormalization(64)
            self.fc = L.Linear(None, n_classes)