small_dnn.py 文件源码

def __call__(self, x):
        self.clear()

        h = self.conv1(x)
        h = F.max_pooling_2d(h, (3, 5), stride=2)

        h = self.conv2(h)

        h = self.conv3(h)
        h = F.max_pooling_2d(h, 3, stride=2, pad=1)

        h = self.conv4(h)

        h = self.fc1(h)
        h = self.fc2(h)

        return h