Menggunakan DataLoader
Kelas DataLoader sangat penting untuk menangani himpunan data berukuran besar secara efisien. Komponen ini mempercepat pelatihan, mengoptimalkan penggunaan memori, dan menstabilkan pembaruan gradien, sehingga membuat model deep learning lebih efektif.
Sekarang, Anda akan membuat DataLoader PyTorch menggunakan dataset dari latihan sebelumnya dan melihat cara kerjanya.
Latihan ini adalah bagian dari kursus
Pengantar Deep Learning dengan PyTorch
Petunjuk latihan
- Impor modul yang diperlukan.
- Buat
DataLoadermenggunakandataset, atur ukuran batch menjadi dua dan aktifkan pengacakan. - Iterasikan
DataLoaderdan cetak setiap batch dari masukan dan label.
Latihan interaktif praktis
Cobalah latihan ini dengan menyelesaikan kode contoh berikut.
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)