code: def __init__(self, size, length):
      self.len = length
      self.data = torch.randn(length, size)
  
file_path: H2O/h2o_flexgen/benchmark/third_party/DeepSpeed/tests/lightning/test_simple.py
imports: import torch
  from pytorch_lightning import LightningModule, Trainer
  from pytorch_lightning.strategies import DeepSpeedStrategy
  from torch.utils.data import DataLoader, Dataset
label_desc: Use image node_id 1 to create a new RandomDataset object from inherited base classes: Dataset with example: obj = RandomDataset(size, length)