net2.py 文件源码

def __call__(self, x, train):
        h = bst.bst(self.b0(self.conv0(x)))
        h = bst.bst(self.b1(self.conv1(h)))
        h = bst.bst(self.b2(self.conv2(h)))
        h = F.max_pooling_2d(h, 2)
        h = F.average_pooling_2d(h, 24)
        h = self.b3(self.fc0(h))
        return h