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

Exercise

Automatic device placement with Accelerator

Your conversational AI model needs to train on a massive dataset, so you've decided to move the model to a GPU. You're leveraging Accelerator for automatic device placement. Note this exercise actually runs on the CPU, but the code remains the same for running on the GPU.

A BERT-based model has been preloaded as model.

Instructions

100 XP
  • Declare an accelerator object by instantiating the appropriate class.
  • Use the accelerator object to prepare the model for distributed training with GPU.