单 HPU 训练

在单个设备上训练与在 Transformers 中一样简单

• 您需要将 Transformers 的 Trainer 类替换为 GaudiTrainer 类，
• 您需要将 Transformers 的 TrainingArguments 类替换为 GaudiTrainingArguments 类，并添加以下参数
  • use_habana 以在 HPU 上执行您的脚本，
  • use_lazy_mode 以使用惰性模式（推荐）或不使用（即，eager 模式），
  • gaudi_config_name 以提供您的 Gaudi 配置文件名称（Hub）或路径（本地）。

要了解更多信息，我们邀请您阅读关于加速训练和预训练的指南。