code: def replace_all_batch_norm_modules_(root: nn.Module) -> nn.Module:
      """
      In place updates :attr:`root` by setting the ``running_mean`` and ``running_var`` to be None and
      setting track_running_stats to be False for any nn.BatchNorm module in :attr:`root`
      """
      # base case
      batch_norm_without_running_stats(root)
  
      for obj in root.modules():
          batch_norm_without_running_stats(obj)
      return root
  
label_desc: Use image node_id 2 for calling a global function with example usage: replace_all_batch_norm_modules_(root) and returns: root