code: def _get_total_batch_size(benchmark_config, model_specs):
      return model_specs["seq_len"] * benchmark_config["batch_size"]
  
file_path: fairscale/benchmarks/datasets/wikitext2_data.py
imports: from collections import namedtuple
  from distutils.version import LooseVersion
  import io
  import operator
  import tempfile
  import torch
  from torch.utils.data import DataLoader
  from torch.utils.data.distributed import DistributedSampler
  import torchtext
  from torchtext.data.utils import get_tokenizer
  from torchtext.utils import download_from_url, extract_archive
label_desc: Use image node_id 2 for calling a global function with example usage: _get_total_batch_size(benchmark_config, model_specs) and returns: unknown