TPUPrecisionPlugin

class pytorch_lightning.plugins.precision.TPUPrecisionPlugin[source]

Bases: pytorch_lightning.plugins.precision.precision_plugin.PrecisionPlugin

Precision plugin for TPU integration.

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

Hook to run the optimizer step.

Return type

Any