code: def get_lifter_after(
      num_joints,
      num_feats=2,
      num_xyz=1,
      num_channels=1024,
      is_generic_baseline=False,
  ):
      output_size = num_joints
      # output_size = num_joints if not is_generic_baseline else num_joints * 3
      input_size = num_joints * num_feats
      return Branch(num_channels, input_size, output_size)
  
file_path: insightface/body/human_pose/ambiguity_aware/lib/models/lifter.py
imports: import torch
  import torch.nn
  from lib.models.building_blocks import Branch, ResBlock
  from lib.utils.misc import init_weights
label_desc: Use image node_id 3 for calling a global function with example usage: get_lifter_after(num_joints, num_feats, num_xyz, num_channels, is_generic_baseline) and returns: Branch