Function hook

ldm/modules/midas/midas/vit.py:160–161

(model, input, output)

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

    return hook