                                     activation="sigmoid", name="face_out")
  
          outputs = [var_x]
  
          if self.config.get("mask_type", False):
              var_y = var_xy  # mask decoder
  
              var_y = self.blocks.upscale2x_hyb(var_y, mask_complexity)
              var_y = self.blocks.upscale2x_hyb(var_y, mask_complexity // 2)
              var_y = self.blocks.upscale2x_hyb(var_y, mask_complexity // 4)
 
</s> remove         if self.config.get("mask_type", False):
 </s> add         if self.config.get("learn_mask", False): </s> remove         if self.config.get("mask_type", False):
 </s> add         if self.config.get("learn_mask", False):
        if self.config.get("learn_mask", False):
                                     activation="sigmoid", name="face_out")
  
          outputs = [var_x]
  
        if self.config.get("learn_mask", False):
              var_y = var_xy  # mask decoder
  
              var_y = self.blocks.upscale2x_hyb(var_y, mask_complexity)
              var_y = self.blocks.upscale2x_hyb(var_y, mask_complexity // 2)
              var_y = self.blocks.upscale2x_hyb(var_y, mask_complexity // 4)
keep keep keep keep replace keep keep keep keep keep
 bugfix: dlight model
change "mask_type" to "learn_mask" in decoders 
plugins/train/model/dlight.py