单 HPU 训练

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

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

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

< > 更新 在 GitHub 上