code: def unfreeze_batch_norm(model):
      for m in model.modules():
          if isinstance(m, torch.nn.BatchNorm2d):
              m.train()
  
label_desc: Use image node_id 2 for calling a global function with example usage: unfreeze_batch_norm(model) without return types