Create a custom dataset

To create a customized dataset in PyTorch, you should create a subclass of the torch.utils.data.Dataset class and implement the following methods: