thanks a lot for the detailed feedback.
A tutorial on writing custom Datasets + Samplers and using transforms seems to be in order.
I’ll be tracking it on https://github.com/pytorch/tutorials/issues 857 and hope to make progress.