TPUPrecisionPlugin

class lightning.pytorch.plugins.precision.TPUPrecisionPlugin(*args, **kwargs)[source]

Bases: PrecisionPlugin

Precision plugin for TPU integration.

optimizer_step(optimizer, model, closure, **kwargs)[source]

Hook to run the optimizer step.

Return type:

Any