class_docstr: A wrapper class to define the common interface used by booster.
  
  Args:
      module (nn.Module): The model to be wrapped.
docstr: Unwrap the model to return the original model for checkpoint saving/loading.
docstr_tok: ["Unwrap","the","model","to","return","the","original","model","for","checkpoint","saving\/loading","."]
code: def unwrap(self):
      """
      Unwrap the model to return the original model for checkpoint saving/loading.
      """
      if isinstance(self.module, ModelWrapper):
          return self.module.unwrap()
      return self.module
  
file_path: colossalai/colossalai/interface/model.py
imports: import torch.nn
label_desc: Use image node_id 2 for calling the ModelWrapper obj's underlying member method code with example usage: obj.unwrap() and returns: self, self