guidedbackprop.py 文件源码

def __init__(self, model):
        super(GuidedBackpropGrad, self).__init__(model)
        for key, funcs in model.predictor.functions.items():
            ismatch = re.match(self.pattern, key)
            if ismatch:
                funcs[-1] = GuidedBackpropReLU()