                  continue
              layer = layer[0]
              layer.set_weights(old_model.get_weights())
          filename = self._io._filename  # pylint:disable=protected-access
          logger.info("Saving Tensorflow 2.x model to '%s'", filename)
          new_model.save(filename)
          # Penalized Loss and Learn Mask used to be disabled automatically if a mask wasn't
 
</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         if self.config["learn_mask"] and self.config["mask_type"] is None:
 </s> add  </s> remove             logger.info("Updating legacy weights from '%s'...", model_name)
 </s> add 
            logger.info("Updating legacy weights from '%s'...", model_name)
                  continue
              layer = layer[0]
            logger.info("Updating legacy weights from '%s'...", model_name)
              layer.set_weights(old_model.get_weights())
          filename = self._io._filename  # pylint:disable=protected-access
          logger.info("Saving Tensorflow 2.x model to '%s'", filename)
          new_model.save(filename)
          # Penalized Loss and Learn Mask used to be disabled automatically if a mask wasn't
keep add keep keep keep keep keep
 Bugfixes:
  - dfl_h128 Legacy Weights update
  - lib.faces_detect - Logging fix 
plugins/train/model/_base.py