code: def get_model_size(args, step_num):
      if step_num == 3:
          return get_model_size(args, 1)
      return getattr(args, f"{model_type[step_num]}_model")
  
file_path: deepspeedexamples/applications/DeepSpeed-Chat/e2e_rlhf.py
imports: import argparse
  import warnings
  import subprocess
  import os
  import datetime
  import time
  import sys
label_desc: Use image node_id 2 for calling a global function with example usage: get_model_size(args, step_num) and returns: getattr, get_model_size