1. Learn
  2. /
  3. Courses
  4. /
  5. Intermediate Deep Learning with PyTorch

Exercise

PyTorch DataLoader

Good job defining the Dataset class! The WaterDataset you just created is now available for you to use.

The next step in preparing the training data is to set up a DataLoader. A PyTorch DataLoader can be created from a Dataset to load data, split it into batches, and perform transformations on the data if desired. Then, it yields a data sample ready for training.

In this exercise, you will build a DataLoader based on the WaterDataset. The DataLoader class you will need has already been imported for you from torch.utils.data. Let's get to it!

Instructions

100 XP
  • Create an instance of WaterDataset from water_train.csv, assigning it to dataset_train.
  • Create dataloader_train based on dataset_train, using a batch size of two and shuffling the samples.
  • Get a batch of features and labels from the DataLoader and print them.