1. Learn
    2. /
  2. Courses
    3. /
  3. Efficient AI Model Training with PyTorch
Connected

Exercise

Prepare datasets for distributed training

You've preprocessed a dataset for a precision agriculture system to help farmers monitor crop health. Now you'll load the data by creating a DataLoader and place the data on GPUs for distributed training, if GPUs are available. Note the exercise actually uses a CPU, but the code is the same for CPUs and GPUs.

Some data has been pre-loaded:

  • A sample dataset with agricultural imagery
  • The Accelerator class from the accelerate library
  • The DataLoader class

Instructions

100 XP
  • Create a dataloader for the pre-defined dataset.
  • Place the dataloader on available devices using the accelerator object.