TPUAccelerator¶
- class pytorch_lightning.accelerators.TPUAccelerator(precision_plugin, training_type_plugin)[source]¶
Bases:
pytorch_lightning.accelerators.accelerator.Accelerator
Accelerator for TPU devices.
- Parameters
precision_plugin¶ (
PrecisionPlugin
) – the plugin to handle precision-specific partstraining_type_plugin¶ (
TrainingTypePlugin
) – the plugin to handle different training routines
- setup(trainer)[source]¶
- Raises
ValueError – If the precision or training type plugin are unsupported.
- Return type