Dataloading on the GPU

Results object
1

Is it possible to perform data preprocessing on the GPU in my data loader without having to copy back over to the CPU?

2

you can do that in training_step of LightningModule for now.