          self._config = config
          self._input_shape = input_shape
  
      @property
      def _model_kwargs(self):
          """ dict: Configuration option for architecture mapped to optional kwargs. """
          return dict(mobilenet=dict(alpha=self._config["mobilenet_width"],
                                     depth_multiplier=self._config["mobilenet_depth"],
                                     dropout=self._config["mobilenet_dropout"]),
                      mobilenet_v2=dict(alpha=self._config["mobilenet_width"]),
 
</s> remove     def __init__(self, input_shape, config):
 </s> add     def __init__(self, input_shape: Tuple[int, int, int], config: dict) -> None: </s> remove     def __init__(self, side, input_shape, config):
 </s> add     def __init__(self,
                 side: Literal["a", "b", "both"],
                 input_shape: Tuple[int, int, int],
                 config: dict) -> None: </s> remove     def _reshape_for_output(self, inputs):
 </s> add     def _reshape_for_output(self, inputs: Tensor) -> Tensor: </s> remove     def __init__(self, side, input_shape, config):
 </s> add     def __init__(self,
                 side: Literal["a", "b", "both", "gblock", "shared"],
                 input_shape: tuple,
                 config: dict) -> None: </s> remove         model = _MODEL_MAPPING.get(arch)
        model["kwargs"] = self._model_kwargs.get(arch, {})
 </s> add         model = _MODEL_MAPPING[arch]
        kwargs = self._model_kwargs.get(arch, {})
    def _model_kwargs(self) -> Dict[str, Dict[str, Union[str, bool]]]:
          self._config = config
          self._input_shape = input_shape
  
      @property
    def _model_kwargs(self) -> Dict[str, Dict[str, Union[str, bool]]]:
          """ dict: Configuration option for architecture mapped to optional kwargs. """
          return dict(mobilenet=dict(alpha=self._config["mobilenet_width"],
                                     depth_multiplier=self._config["mobilenet_depth"],
                                     dropout=self._config["mobilenet_dropout"]),
                      mobilenet_v2=dict(alpha=self._config["mobilenet_width"]),
keep keep keep keep replace keep keep keep keep keep
 Model updates
- Increase model summary width
- Phaze A updates
  - Update some min/max values
  - Add Decoder Filter Slope Mode
  - Add additional arguments for Upsampling2D
  - Adjust upsampling method for multiple upsamples in FC layers
  - Typing 
plugins/train/model/phaze_a.py