code: def _set_wa_wb(self):
      if self._model is not None:
          if hasattr(self, "wa") and hasattr(self, "wb"):
              self._model.aggregator.wa.share = self.wa
              self._model.aggregator.wb.share = self.wb
  
imports: from typing import Union, Literal, Tuple
  import torch
  from fate.arch import Context
  from fate.ml.nn.model_zoo.agg_layer.agg_layer import _AggLayerBase
  from fate.arch.protocol.mpc.nn.sshe.nn_layer import SSHENeuralNetworkAggregatorLayer, SSHENeuralNetworkOptimizerSGD
label_desc: Use image node_id 3 for calling the SSHEAggLayerHost obj's underlying member method code with example usage: obj._set_wa_wb() without return types