Uso de DataLoader
La clase DataLoader
es esencial para manejar eficazmente grandes conjuntos de datos. Acelera el entrenamiento, optimiza el uso de memoria y estabiliza las actualizaciones de gradiente, haciendo que los modelos de aprendizaje profundo sean más eficaces.
Ahora, crearás un PyTorch DataLoader
utilizando el dataset
del ejercicio anterior y lo verás en acción.
Este ejercicio forma parte del curso
Introducción al aprendizaje profundo con PyTorch
Instrucciones del ejercicio
- Importa el módulo necesario.
- Crea un
DataLoader
utilizandodataset
, estableciendo un tamaño de lote de dos y activando el barajado. - Recorre la página
DataLoader
e imprime cada lote de entradas y etiquetas.
Ejercicio interactivo práctico
Prueba este ejercicio completando el código de muestra.
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)