def __init__()

in pai-python-sdk/training/pytorch_ddp/train_src/train_multinode.py [0:0]


    def __init__(self, size):
        self.size = size
        self.data = [(torch.rand(20), torch.rand(1)) for _ in range(size)]