TPUPrecisionPlugin¶

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

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

Hook to run the optimizer step.