Siapkan himpunan data untuk pelatihan terdistribusi

Anda telah melakukan prapemrosesan sebuah himpunan data untuk sistem pertanian presisi guna membantu petani memantau kesehatan tanaman. Sekarang Anda akan memuat data dengan membuat sebuah DataLoader dan menempatkan data pada GPU untuk pelatihan terdistribusi, jika GPU tersedia. Perhatikan bahwa latihan ini sebenarnya menggunakan CPU, tetapi kodenya sama untuk CPU dan GPU.

Sebagian data telah dimuat sebelumnya:

Sebuah dataset contoh dengan citra pertanian
Kelas Accelerator dari pustaka accelerate
Kelas DataLoader

Latihan ini adalah bagian dari kursus

Pelatihan Model AI Efisien dengan PyTorch

Lihat Kursus

Petunjuk latihan

Buat dataloader untuk dataset yang telah didefinisikan.
Tempatkan dataloader pada perangkat yang tersedia menggunakan objek accelerator.

Latihan interaktif praktis

Cobalah latihan ini dengan menyelesaikan kode contoh berikut.

accelerator = Accelerator()

# Create a dataloader for the pre-defined dataset
dataloader = ____(____, batch_size=32, shuffle=True)

# Place the dataloader on available devices
dataloader = accelerator.____(____)

print(accelerator.device)

Edit dan Jalankan Kode