code: def __init__(self, gpu0_bsz, *args, **kwargs):
      self.gpu0_bsz = gpu0_bsz
      super().__init__(*args, **kwargs)
  
file_path: thu-pacman-faster-moe/examples/transformer-xl/utils/data_parallel.py
imports: from torch.nn.parallel import DataParallel
  import torch
  from torch.nn.parallel._functions import Scatter
  from torch.nn.parallel.parallel_apply import parallel_apply
label_desc: Use image node_id 1 to create a new BalancedDataParallel object from inherited base classes: DataParallel with example: obj = BalancedDataParallel(gpu0_bsz)