docstr: Returns the learned model(s).
docstr_tok: ["Returns","the","learned","model","(","s",")","."]
code: def get_model(self):
      """Returns the learned model(s)."""
      state = self.get_state_dict()
      model = self.model_creator(self.config)
      model_state = state["models"][0]
      model.load_state_dict(model_state)
      return model.module if hasattr(model, "module") else model
  
imports: import io
  import types
  import logging
  import numbers
  import torch
  import numpy
  from zoo.orca.data.ray_xshards import RayXShards
  from zoo.orca.learn.pytorch.training_operator import TrainingOperator
  from zoo.orca.learn.pytorch.torch_runner import TorchRunner
  from zoo.orca.learn.utils import maybe_dataframe_to_xshards, dataframe_to_xshards, convert_predict_xshards_to_dataframe, update_predict_xshards, process_xshards_of_pandas_dataframe
  from zoo.ray import RayContext
  import ray
  from ray.exceptions import RayActorError
label_desc: Use image node_id 8 for calling the PyTorchRayEstimator obj's underlying member method code with example usage: obj.get_model() and returns: unknown