      """
      def __init__(self, plugin):
          logger.debug("Initializing %s: (plugin: %s)", self.__class__.__name__, plugin)
          self._model = plugin.model
          self._name = plugin.name
          self._do_freeze = plugin._args.freeze_weights
          self._weights_file = self._check_weights_file(plugin._args.load_weights)
  
          freeze_layers = plugin.config.get("freeze_layers")  # Standardized config for freezing
          load_layers = plugin.config.get("load_layers")  # Standardized config for loading
 
</s> remove         autoencoder = KerasModel(inputs, outputs, name=self.name)
 </s> add         autoencoder = KerasModel(inputs, outputs, name=self.model_name) </s> remove         autoencoder = KerasModel(inputs, outputs, name=self.name)
 </s> add         autoencoder = KerasModel(inputs, outputs, name=self.model_name) </s> remove         autoencoder = KerasModel(inputs, outputs, name=self.name)
 </s> add         autoencoder = KerasModel(inputs, outputs, name=self.model_name) </s> remove         autoencoder = KerasModel(inputs, outputs, name=self.name)
 </s> add         autoencoder = KerasModel(inputs, outputs, name=self.model_name)
        self._name = plugin.model_name
      """
      def __init__(self, plugin):
          logger.debug("Initializing %s: (plugin: %s)", self.__class__.__name__, plugin)
          self._model = plugin.model
        self._name = plugin.model_name
          self._do_freeze = plugin._args.freeze_weights
          self._weights_file = self._check_weights_file(plugin._args.load_weights)
  
          freeze_layers = plugin.config.get("freeze_layers")  # Standardized config for freezing
          load_layers = plugin.config.get("load_layers")  # Standardized config for loading
keep keep keep keep replace keep keep keep keep keep
 Bugfix -Weights freezing/loading for dfl-sae 
plugins/train/model/_base.py