          # Penalized Loss and Learn Mask used to be disabled automatically if a mask wasn't
          # selected, so disable it if enabled, but mask_type is None
          if self.config["penalized_mask_loss"] and self.config["mask_type"] is None:
              self.config["penalized_mask_loss"] = False
          if self.config["learn_mask"] and self.config["mask_type"] is None:
              self.config["learn_mask"] = False
          self._state.save()
  
      def _validate_input_shape(self):
          """ Validate that the input shape is either a single shape tuple of 3 dimensions or
 
</s> remove         if self.config["penalized_mask_loss"] and self.config["mask_type"] is None:
 </s> add         if self.config["mask_type"] is None: </s> remove             logger.info("Updating legacy weights from '%s'...", model_name)
 </s> add 
          # Penalized Loss and Learn Mask used to be disabled automatically if a mask wasn't
          # selected, so disable it if enabled, but mask_type is None
          if self.config["penalized_mask_loss"] and self.config["mask_type"] is None:
              self.config["penalized_mask_loss"] = False

              self.config["learn_mask"] = False
          self._state.save()
  
      def _validate_input_shape(self):
          """ Validate that the input shape is either a single shape tuple of 3 dimensions or
keep keep keep keep replace keep keep keep keep keep
 Bugfixes:
  - dfl_h128 Legacy Weights update
  - lib.faces_detect - Logging fix 
plugins/train/model/_base.py