Datasets - how to add augmentations?

Hi. I want to add random augmentations to my dataset. Meaning that I do not want to perform augmentations once during the creating of the dataset, but rather to generate the augmentation randomly during training. This is usually implemented in getitem method of default pytorch dataset class. Is it possible to make this functionality with current version of Datasets package?

You can use set_transform for this.