code: def get_model_obj(model):
      model = model.module if hasattr(model, "module") else model
      return model
  
imports: from torch import nn
  import torch
label_desc: Use image node_id 1 for calling a global function with example usage: get_model_obj(model) and returns: model