TPUPrecisionPlugin¶
- class lightning.pytorch.plugins.precision.TPUPrecisionPlugin(*args, **kwargs)[source]¶
Bases:
lightning.pytorch.plugins.precision.precision_plugin.PrecisionPlugin
Precision plugin for TPU integration.
Bases: lightning.pytorch.plugins.precision.precision_plugin.PrecisionPlugin
Precision plugin for TPU integration.