simple_model.py 文件源码

def functions(self):
        return collections.OrderedDict([
            ('conv1', [self.conv1, self.bnorm1, F.relu]),
            ('pool1', [_max_pooling_2d]),
            ('conv2', [self.conv2, self.bnorm2, F.relu]),
            ('pool2', [_spatial_pyramid_pooling_2d]),
            ('fc', [self.fc])
        ])