code: def __init__(self, name: str, save_callback: Callable):
      self.save_callback = save_callback
      self.name = name
  
imports: from typing import Callable, Dict, List, TYPE_CHECKING
  from labml.internal.util.strings import is_pattern_match
label_desc: Use image node_id 1 to create a new ForwardHook object  with example: obj = ForwardHook(name, save_callback)