      """ SAE Model from DFL """
      def __init__(self, *args, **kwargs):
          super().__init__(*args, **kwargs)
          self.input_shape = (self.config["input_size"], self.config["input_size"], 3)
          self.architecture = self.config["architecture"].lower()
          self.use_mask = self.config.get("learn_mask", False)
          self.multiscale_count = 3 if self.config["multiscale_decoder"] else 1
          self.encoder_dim = self.config["encoder_dims"]
          self.decoder_dim = self.config["decoder_dims"]
  
 
</s> remove from ._base import ModelBase, KerasModel
 </s> add from ._base import ModelBase, KerasModel, logger </s> remove                                  name="{}_{}".format(self.name, self.architecture))
 </s> add                                  name=self.name) </s> remove         The parent plugin class that owns the IO functions.
 </s> add         The parent plugin class that owns the weights functions.
      """ SAE Model from DFL """
      def __init__(self, *args, **kwargs):
          super().__init__(*args, **kwargs)
          self.input_shape = (self.config["input_size"], self.config["input_size"], 3)

          self.use_mask = self.config.get("learn_mask", False)
          self.multiscale_count = 3 if self.config["multiscale_decoder"] else 1
          self.encoder_dim = self.config["encoder_dims"]
          self.decoder_dim = self.config["decoder_dims"]
  
keep keep keep keep replace keep keep keep keep keep
 Bugfix -Weights freezing/loading for dfl-sae 
plugins/train/model/dfl_sae.py