code: def freeze_bn(net, use_global_stats=True):
      """Freeze BatchNorm layers by setting `use_global_stats` to `True`
  
      Parameters
      ----------
      net : mxnet.gluon.Block
          The network whose BatchNorm layers are going to be modified
      use_global_stats : bool
          The value of `use_global_stats` to set for all BatchNorm layers
  
      Returns
      ------
      mxnet.gluon.Block
          Original network with BatchNorm layers modified.
  
      """
      recursive_visit(
          net, _freeze_bn_callback, use_global_stats=use_global_stats
      )
  
imports: import re
  import numpy
label_desc: Use image node_id 4 for calling a global function with example usage: freeze_bn(net, use_global_stats) without return types