1. 学习
    2. /
  2. 课程
    3. /
  3. Efficient AI Model Training with PyTorch
Connected

练习

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.

说明

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