If we use a combination of the Dataset and Dataloader classes (as shown below), I have to explicitly load the data onto the GPU using
Dataset
Dataloader