activation_hooks

Modules

modelopt.torch.puzzletron.activation_scoring.activation_hooks.utils

Provides a function to register activation hooks for a model.