  
          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)
              var_y = self.blocks.upscale2x_hyb(var_y, mask_complexity // 8)
  
              var_y = self.blocks.conv2d(var_y, 1, kernel_size=5, padding="same",
                                         activation="sigmoid", name="mask_out")
  
              outputs.append(var_y)
 
</s>
            var_y = self.blocks.upscale2x(var_y, mask_complexity, fast=False)
            var_y = self.blocks.upscale2x(var_y, mask_complexity // 2, fast=False)
            var_y = self.blocks.upscale2x(var_y, mask_complexity // 4, fast=False)
            var_y = self.blocks.upscale2x(var_y, mask_complexity // 8, fast=False)
  
          if self.config.get("learn_mask", False):
              var_y = var_xy  # mask decoder
  
            var_y = self.blocks.upscale2x(var_y, mask_complexity, fast=False)
            var_y = self.blocks.upscale2x(var_y, mask_complexity // 2, fast=False)
            var_y = self.blocks.upscale2x(var_y, mask_complexity // 4, fast=False)
            var_y = self.blocks.upscale2x(var_y, mask_complexity // 8, fast=False)
            var_y = self.blocks.upscale2x(var_y, mask_complexity, fast=False)
            var_y = self.blocks.upscale2x(var_y, mask_complexity // 2, fast=False)
            var_y = self.blocks.upscale2x(var_y, mask_complexity // 4, fast=False)
            var_y = self.blocks.upscale2x(var_y, mask_complexity // 8, fast=False)
            var_y = self.blocks.upscale2x(var_y, mask_complexity, fast=False)
            var_y = self.blocks.upscale2x(var_y, mask_complexity // 2, fast=False)
            var_y = self.blocks.upscale2x(var_y, mask_complexity // 4, fast=False)
            var_y = self.blocks.upscale2x(var_y, mask_complexity // 8, fast=False)
            var_y = self.blocks.upscale2x(var_y, mask_complexity, fast=False)
            var_y = self.blocks.upscale2x(var_y, mask_complexity // 2, fast=False)
            var_y = self.blocks.upscale2x(var_y, mask_complexity // 4, fast=False)
            var_y = self.blocks.upscale2x(var_y, mask_complexity // 8, fast=False)
  
              var_y = self.blocks.conv2d(var_y, 1, kernel_size=5, padding="same",
                                         activation="sigmoid", name="mask_out")
  
              outputs.append(var_y)
keep keep keep keep replace replace replace replace keep keep keep keep keep
 Move custom dlight blocks to nn_blocks
plugins/train/model/dlight.py