code: def get_model(self):
      return {
          "optimizer": self.optimizer.state_dict(),
          "lr_scheduler": self.lr_scheduler.state_dict(),
          "end_epoch": self.end_epoch,
          "is_converged": self.is_converged,
          "tol": self.tol,
          "early_stop": self.early_stop,
      }
  
imports: import logging
  import torch
  from fate.arch import Context
  from fate.arch.dataframe import DataLoader
  from fate.ml.abc.module import HeteroModule
  from fate.ml.utils._convergence import converge_func_factory
  from fate.ml.utils._optimizer import LRScheduler, Optimizer, separate
label_desc: Use image node_id 3 for calling the HeteroLinREstimatorArbiter obj's underlying member method code with example usage: obj.get_model() and returns: dict