單 HPU 訓練

在單個裝置上進行訓練與在 Transformers 中一樣簡單

• 您需要將 Transformers 的 Trainer 類替換為 GaudiTrainer 類，
• 您需要將 Transformers 的 TrainingArguments 類替換為 GaudiTrainingArguments 類，並新增以下引數
  • use_habana 用於在 HPU 上執行指令碼，
  • use_lazy_mode 用於使用延遲模式（推薦）或不使用延遲模式（即，急切模式），
  • gaudi_config_name 用於提供 Gaudi 配置檔案（Hub 或本地路徑）的名稱。

為了進一步瞭解，我們邀請您閱讀我們關於加速訓練和預訓練的指南。