Table of Contents

Shortcuts

TPUHalfPrecisionPlugin¶

class pytorch_lightning.plugins.precision.TPUHalfPrecisionPlugin[source]¶

Bases: pytorch_lightning.plugins.precision.precision_plugin.PrecisionPlugin

Plugin that enables bfloats on TPUs

connect(model, optimizers, lr_schedulers)[source]¶

Connects this plugin to the accelerator and the training process

Return type: Tuple[Module, List[Optimizer], List[Any]]