Function hook

dmidas/backbones/utils.py:57–58

(model, input, output)

def get_activation(name):
    def hook(model, input, output):
        activations[name] = output

    return hook