code: def set_batch_size(self, batch_size):
      self.batch_size = batch_size
      self.estimator.batch_size = batch_size
  
imports: import logging
  import torch
  from fate.arch import Context, dataframe
  from fate.arch.dataframe import DataFrame
  from fate.arch.protocol.mpc.nn.sshe.lr_layer import SSHELogisticRegressionLayer, SSHELogisticRegressionLossLayer, SSHEOptimizerSGD
  from fate.ml.abc.module import Module, HeteroModule
  from fate.ml.utils import predict_tools
  from fate.ml.utils._convergence import converge_func_factory
  from fate.ml.utils._model_param import get_initialize_func
  from fate.ml.utils._model_param import serialize_param, deserialize_param
label_desc: Use image node_id 2 for calling the SSHELogisticRegression obj's underlying member method code with example usage: obj.set_batch_size(batch_size) without return types