def METHOD_NAME(self) -> Dict[str, Dict[str, torch.Tensor]]:
    data = defaultdict(dict)
    for module_name, hooks in self.hooks.items():
        for target_name, hook in hooks.items():
            if len(hook.buffer) > 0:
                data[module_name][target_name] = hook.buffer[0] / self.training_steps
    return data