base.py 文件源码

def __init__(self, output_layer):
        self.output_layer = output_layer
        self.layers = self._collect_layers()
        self.parameters = L.get_all_param_values(self.output_layer)
        self.stats_f = None