Method __init__

diff2flow/dataloader.py:265–272

kwargs must contain the output keys with their corresponding shapes, e.g. image=(3, 32, 32), label=(1,)

(self, num_samples: int = 10000000, **kwargs)

class DummyDataset(Dataset):
    def __init__(self, num_samples: int = 10000000, **kwargs):
        """
        kwargs must contain the output keys with their corresponding shapes,
        e.g. image=(3, 32, 32), label=(1,)
        """
        super().__init__()
        self.num_samples = num_samples
        self.kwargs = {k: tuple(v) if not isinstance(v, str) else v for k, v in kwargs.items()}

    def __len__(self):
        return int(self.num_samples)