          if self.config.get("learn_mask", False):
              var_y = var_xy  # mask decoder
  
              var_y = Upscale2xBlock(mask_complexity, activation="leakyrelu", fast=False)(var_y)
              var_y = Upscale2xBlock(mask_complexity // 2, activation="leakyrelu", fast=False)(var_y)
              var_y = Upscale2xBlock(mask_complexity // 4, activation="leakyrelu", fast=False)(var_y)
              var_y = Upscale2xBlock(mask_complexity // 8, activation="leakyrelu", fast=False)(var_y)
 
</s>
            var_y = LeakyReLU(alpha=0.1)(var_y)
          if self.config.get("learn_mask", False):
              var_y = var_xy  # mask decoder
            var_y = LeakyReLU(alpha=0.1)(var_y)
  
              var_y = Upscale2xBlock(mask_complexity, activation="leakyrelu", fast=False)(var_y)
              var_y = Upscale2xBlock(mask_complexity // 2, activation="leakyrelu", fast=False)(var_y)
              var_y = Upscale2xBlock(mask_complexity // 4, activation="leakyrelu", fast=False)(var_y)
              var_y = Upscale2xBlock(mask_complexity // 8, activation="leakyrelu", fast=False)(var_y)
keep add keep keep keep keep keep
 Fixups:
  - lib.gui.display - Catch error when trying to delete non-existant notebook tabs
  - lib.model.nn_blocks.ResidualBlock: Remove first LeakyReLU (explicit > implicit)
  - lib.model.nn_blocks._get_default_initializer - Always use ConvAware if selected
  - plugins.model.model - Add explicit leakyReLUs for ResBlocks
plugins/train/model/dlight.py