Dear Thomas, thanks again for the quick reply, really appreciate it.
In the meanwhile, I found this post 2. Apparently I can simply change the dataset.
The torchvision datasets seem to contain the attribute self.data which I will try to backup in a different variable and overwrite by a subset of the original data. Hope this will lead to an efficient training on a smaller training set.
Will get back to you once I see how this goes, in case you have any concerns please let me know!
Best,
Max