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
Exercise instructions
- Import the required module.
- Create a
DataLoader
usingdataset
, 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)