class_docstr: A wrapper class to define the common interface used by booster.
  
  Args:
      module (nn.Module): The model to be wrapped.
code: def __init__(self, module: nn.Module) -> None:
      super().__init__()
      self.module = module
  
file_path: ColossalAI/colossalai/interface/model.py
imports: import torch.nn
label_desc: Use image node_id 1 to create a new ModelWrapper object from inherited base classes: nn with example: obj = ModelWrapper(module)