Method hook_fn

slowfast/visualization/utils.py:198–199

(module, input, output)

        """

        def hook_fn(module, input, output):
            self.hooks[layer_name] = output.clone().detach()

        layer = get_layer(self.model, layer_name)
        layer.register_forward_hook(hook_fn)