Error when loading the datasets

[Kabongosalomon]

Hi, thank you for the clean code.

I'm having issues, trying to run this cell:

root = "./datasets"

train_sampler = None

if args.dataset == "STL10":
    train_dataset = torchvision.datasets.STL10(
        root, split="unlabeled", download=True, transform=TransformsSimCLR()
    )
elif args.dataset == "CIFAR10":
    train_dataset = torchvision.datasets.CIFAR10(
        root, download=True, transform=TransformsSimCLR()
    )
else:
    raise NotImplementedError

train_loader = torch.utils.data.DataLoader(
    train_dataset,
    batch_size=args.batch_size,
    shuffle=(train_sampler is None),
    drop_last=True,
    num_workers=args.workers,
    sampler=train_sampler,
)

This is the error message, I'm getting :

---------------------------------------------------------------------------
TypeError                                 Traceback (most recent call last)
<ipython-input-13-bc54d6807dbe> in <module>
      9 elif args.dataset == "CIFAR10":
     10     train_dataset = torchvision.datasets.CIFAR10(
---> 11         root, download=True, transform=TransformsSimCLR()
     12     )
     13 else:

TypeError: __init__() missing 1 required positional argument: 'size'

[Spijkervet]

Thanks for letting me know, I adjusted the notebook to comply with a more recent version of the repository :)