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
  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 2 for calling the CoordinatedLinRModuleArbiter obj's underlying member method code with example usage: obj.set_batch_size(batch_size) without return types