Como usar o DataLoader
A classe DataLoader
é essencial para o manuseio eficiente de grandes conjuntos de dados. Ela agiliza o treinamento, otimiza o uso da memória e estabiliza as atualizações de gradiente, tornando os modelos de aprendizado profundo mais eficazes.
Agora, você vai criar um DataLoader
PyTorch usando o dataset
do exercício anterior e vê-lo em ação.
Este exercício faz parte do curso
Introdução ao Aprendizado Profundo com o PyTorch
Instruções do exercício
- Importe o módulo necessário.
- Crie um
DataLoader
usandodataset
, definindo um tamanho de lote igual a dois e ativando o embaralhamento (shuffling). - Itere pelo
DataLoader
e imprima cada lote de entradas e rótulos.
Exercício interativo prático
Experimente este exercício completando este código de exemplo.
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)