Get startedGet started for free

Using DataLoader

The DataLoader class is essential for efficiently handling large datasets. It speeds up training, optimizes memory usage, and stabilizes gradient updates, making deep learning models more effective.

Now, you'll create a PyTorch DataLoader using the dataset from the previous exercise and see it in action.

This exercise is part of the course

Introduction to Deep Learning with PyTorch

View Course

Exercise instructions

  • Import the required module.
  • Create a DataLoader using dataset, setting a batch size of two and enabling shuffling.
  • Iterate through the DataLoader and print each batch of inputs and labels.

Hands-on interactive exercise

Have a go at this exercise by completing this sample code.

from torch.utils.data import ____

# Create a DataLoader
dataloader = ____

# Iterate over the dataloader
for batch_inputs, batch_labels in dataloader:
    print('batch_inputs:', batch_inputs)
    print('batch_labels:', batch_labels)
Edit and Run Code