code: def u2net_lite(out_ch: int = 1):
      cfg = {
          # height, in_ch, mid_ch, out_ch, RSU4F, side
          "encode": [
              [7, 3, 16, 64, False, False],  # En1
              [6, 64, 16, 64, False, False],  # En2
              [5, 64, 16, 64, False, False],  # En3
              [4, 64, 16, 64, False, False],  # En4
              [4, 64, 16, 64, True, False],  # En5
              [4, 64, 16, 64, True, True],
          ],  # En6
          # height, in_ch, mid_ch, out_ch, RSU4F, side
          "decode": [
              [4, 128, 16, 64, True, True],  # De5
              [4, 128, 16, 64, False, True],  # De4
              [5, 128, 16, 64, False, True],  # De3
              [6, 128, 16, 64, False, True],  # De2
              [7, 128, 16, 64, False, True],
          ],  # De1
      }
  
      return U2Net(cfg, out_ch)
  
imports: from typing import Union, List
  import torch
  import torch.nn
  import torch.nn.functional
label_desc: Use image node_id 2 for calling a global function with example usage: u2net_lite(out_ch) and returns: U2Net