MulaiMulai sekarang secara gratis

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

Lihat Kursus

Petunjuk latihan

  • Impor modul yang diperlukan.
  • Buat DataLoader menggunakan dataset, atur ukuran batch menjadi dua dan aktifkan pengacakan.
  • Iterasikan DataLoader dan 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)
Edit dan Jalankan Kode