Class DummyDataset

trainer/__init__.py:14–34

    from torch.utils.data import DataLoader, Dataset

    class DummyDataset(Dataset):

        def __init__(self) -> None:
            super().__init__()
            data = torch.randn(
                1,
                3,
                1024,
                1024,
            )
            self.data = data.expand(100000, -1, -1, -1)

        def __len__(self, ):
            return len(self.data)

        def __getitem__(self, index):
            return (
                self.data[index],
                self.data[index],
                'A cat holding a sign that says hello world',
            )

    dataset = DummyDataset()
    train_loader = DataLoader(