IPUAccelerator¶
- class pytorch_lightning.accelerators.IPUAccelerator(precision_plugin, training_type_plugin)[source]¶
Bases:
pytorch_lightning.accelerators.accelerator.Accelerator
Accelerator for IPUs.
- Parameters
precision_plugin¶ (
PrecisionPlugin
) – the plugin to handle precision-specific partstraining_type_plugin¶ (
TrainingTypePlugin
) – the plugin to handle different training routines