code: def set_bn_momentum_default(bn_momentum):
      def fn(m):
          if isinstance(
              m, (nn.BatchNorm1d, nn.BatchNorm2d, nn.BatchNorm3d)
          ):
              m.momentum = bn_momentum
  
      return fn
  
imports: import numpy
  import matplotlib.pyplot
  from mpl_toolkits.mplot3d import Axes3D
  import random
  import torch
  import torch.nn
  import torch.nn.functional
  import os
  from collections import abc
label_desc: Use image node_id 8 for calling a global function with example usage: set_bn_momentum_default(bn_momentum) and returns: fn