code: def get_norm(zero_init=False):
      if cfg.BACKBONE.NORM == "None":
          return lambda x: x
      if cfg.BACKBONE.NORM == "GN":
          Norm = GroupNorm
          layer_name = "gn"
      else:
          Norm = BatchNorm
          layer_name = "bn"
      return lambda x: Norm(
          layer_name,
          x,
          gamma_initializer=tf.zeros_initializer()
          if zero_init
          else None,
      )
  
imports: import numpy
  import tensorflow
  from contextlib import ExitStack, contextmanager
  from tensorpack import tfv1
  from tensorpack.models import BatchNorm, Conv2D, MaxPooling, layer_register
  from tensorpack.tfutils import argscope
  from tensorpack.tfutils.scope_utils import auto_reuse_variable_scope
  from tensorpack.tfutils.varreplace import custom_getter_scope, freeze_variables
  from config import config
label_desc: Use image node_id 6 for calling a global function with example usage: get_norm(zero_init) and returns: function, function