Why are transforms given to the dataset and not to the dataloader?

The get_sample() function is actually a cool workaround. I’ll try that coupled with a transforms argument for a custom data loader.

Thanks for your time!