code: def get_total_num_stages(**kwargs):
      """Returns total number of training stages."""
      return 2 * kwargs["num_resolutions"] - 1
  
imports: import os
  import time
  from absl import logging
  from magenta.models.gansynth.lib import networks
  import numpy
  import tensorflow.compat.v1
  import tensorflow_gan
label_desc: Use image node_id 4 for calling a global function with example usage: get_total_num_stages() and returns: unknown