          var_x = self.blocks.upscale2x_hyb(var_x, decoder_b_complexity)
          var_x = self.blocks.res_block(var_x, decoder_b_complexity, use_bias=True)
          var_x = self.blocks.res_block(var_x, decoder_b_complexity, use_bias=False)
          var_x = BatchNormalization()(var_x)
          var_x = self.blocks.upscale2x_hyb(var_x, decoder_b_complexity // 2)
          var_x = self.blocks.res_block(var_x, decoder_b_complexity // 2, use_bias=True)
          var_x = self.blocks.upscale2x_hyb(var_x, decoder_b_complexity // 4)
          var_x = self.blocks.res_block(var_x, decoder_b_complexity // 4, use_bias=False)
          var_x = BatchNormalization()(var_x)
          var_x = self.blocks.upscale2x_hyb(var_x, decoder_b_complexity // 8)
 
</s> remove         var_x = self.blocks.upscale2x_hyb(var_x, decoder_b_complexity)
 </s> add         var_x = self.blocks.upscale2x(var_x, decoder_b_complexity, fast=False) </s> remove         var_x = self.blocks.upscale2x_hyb(var_x, decoder_b_complexity // 4)
 </s> add         var_x = self.blocks.upscale2x(var_x, decoder_b_complexity // 4, fast=False) </s> remove         var_x = self.blocks.upscale2x_hyb(var_x, decoder_b_complexity // 8)
 </s> add         var_x = self.blocks.upscale2x(var_x, decoder_b_complexity // 8, fast=False) </s> remove         var_x = self.blocks.upscale2x_fast(var_x, decoder_b_complexity)
        var_x = self.blocks.upscale2x_fast(var_x, decoder_b_complexity // 2)
        var_x = self.blocks.upscale2x_fast(var_x, decoder_b_complexity // 4)
        var_x = self.blocks.upscale2x_fast(var_x, decoder_b_complexity // 8)
 </s> add         var_x = self.blocks.upscale2x(var_x, decoder_b_complexity, fast=True)
        var_x = self.blocks.upscale2x(var_x, decoder_b_complexity // 2, fast=True)
        var_x = self.blocks.upscale2x(var_x, decoder_b_complexity // 4, fast=True)
        var_x = self.blocks.upscale2x(var_x, decoder_b_complexity // 8, fast=True) </s> remove         var_x = self.blocks.upscale2x_hyb(var_x, decoder_a_complexity)
        var_x = self.blocks.upscale2x_hyb(var_x, decoder_a_complexity // 2)
        var_x = self.blocks.upscale2x_hyb(var_x, decoder_a_complexity // 4)
        var_x = self.blocks.upscale2x_hyb(var_x, decoder_a_complexity // 8)
 </s> add         var_x = self.blocks.upscale2x(var_x, decoder_a_complexity, fast=False)
        var_x = self.blocks.upscale2x(var_x, decoder_a_complexity // 2, fast=False)
        var_x = self.blocks.upscale2x(var_x, decoder_a_complexity // 4, fast=False)
        var_x = self.blocks.upscale2x(var_x, decoder_a_complexity // 8, fast=False)
        var_x = self.blocks.upscale2x(var_x, decoder_b_complexity // 2, fast=False)
          var_x = self.blocks.upscale2x_hyb(var_x, decoder_b_complexity)
          var_x = self.blocks.res_block(var_x, decoder_b_complexity, use_bias=True)
          var_x = self.blocks.res_block(var_x, decoder_b_complexity, use_bias=False)
          var_x = BatchNormalization()(var_x)
        var_x = self.blocks.upscale2x(var_x, decoder_b_complexity // 2, fast=False)
          var_x = self.blocks.res_block(var_x, decoder_b_complexity // 2, use_bias=True)
          var_x = self.blocks.upscale2x_hyb(var_x, decoder_b_complexity // 4)
          var_x = self.blocks.res_block(var_x, decoder_b_complexity // 4, use_bias=False)
          var_x = BatchNormalization()(var_x)
          var_x = self.blocks.upscale2x_hyb(var_x, decoder_b_complexity // 8)
keep keep keep keep replace keep keep keep keep keep
 Move custom dlight blocks to nn_blocks 
plugins/train/model/dlight.py