code: def get_model(model:nn.Module):
      "Return the model maybe wrapped inside `model`."
      return model.module if isinstance(model, (DistributedDataParallel, nn.DataParallel)) else modef get_model(model:nn.Module):
      "Return the model maybe wrapped inside `model`."
      return model.module if isinstance(model, (DistributedDataParallel, nn.DataParallel)) else model
docs: Return the model maybe wrapped inside `model`.