code: def __init__(self, batch_size):
      super().__init__()
      self.batch_size = batch_size
      self.bn_layer = nn.BatchNorm1d(1)
      self.linear = nn.Linear(1, 10)
      self.bn_outputs = []
  
file_path: pytorch-lightning/tests/parity_pytorch/test_sync_batchnorm_parity.py
imports: import torch
  import torch.nn
  from lightning.pytorch import LightningModule, Trainer, seed_everything
  from torch.utils.data import DataLoader, DistributedSampler
  from parity_pytorch import RunIf
label_desc: Use image node_id 1 to create a new SyncBNModule object from inherited base classes: LightningModule with example: obj = SyncBNModule(batch_size)