> For the complete documentation index, see [llms.txt](https://boinc-ai.gitbook.io/optimum/llms.txt). Markdown versions of documentation pages are available by appending `.md` to page URLs; this page is available as [Markdown](https://boinc-ai.gitbook.io/optimum/habana/tutorials/single-hpu-training.md).

# Single-HPU Training

## Single-HPU Training

Training on a single device is as simple as in Transformers:

* You need to replace the Transformers’ [`Trainer`](https://huggingface.co/docs/transformers/main_classes/trainer) class with the [`GaudiTrainer`](https://huggingface.co/docs/optimum/habana/package_reference/trainer) class,
* You need to replace the Transformers’ [`TrainingArguments`](https://huggingface.co/docs/transformers/main_classes/trainer#transformers.TrainingArguments) class with the [GaudiTrainingArguments](https://huggingface.co/docs/optimum.habana/main/en/package_reference/trainer#optimum.habana.GaudiTrainingArguments) class and add the following arguments:
  * `use_habana` to execute your script on an HPU,
  * `use_lazy_mode` to use lazy mode (recommended) or not (i.e. eager mode),
  * `gaudi_config_name` to give the name of (Hub) or the path to (local) your Gaudi configuration file.

To go further, we invite you to read our guides about [accelerating training](https://huggingface.co/docs/optimum/habana/usage_guides/accelerate_training) and [pretraining](https://huggingface.co/docs/optimum/habana/usage_guides/pretraining).