Single-HPU Training
Last updated
Last updated
Training on a single device is as simple as in Transformers:
You need to replace the Transformers’ class with the class,
You need to replace the Transformers’ class with the 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 and .