code: def loss_visualize(loss):
      """loss_visualize"""
      visualize_loss = static.create_global_var(
          persistable=True, dtype="float32", shape=[1], value=0
      )
      batch_count = static.create_global_var(
          persistable=True, dtype="float32", shape=[1], value=0
      )
      inner_add(loss, visualize_loss)
      inner_add(1.0, batch_count)
  
      return visualize_loss, batch_count
  
imports: import os
  import sys
  import math
  import time
  import numpy
  import pickle
  import paddle
  import paddle.fluid
  import paddle.fluid.layers
  import paddle.static
  from paddle.common_ops_import import LayerHelper, check_type, check_variable_and_dtype
  from paddle.fluid.framework import Variable
  import pgl
  from pgl.utils.logger import log
label_desc: Use image node_id 5 for calling a global function with example usage: loss_visualize(loss) and returns: visualize_loss,  batch_count