SingleTPUStrategy

class lightning_fabric.strategies.SingleTPUStrategy(device, accelerator=None, checkpoint_io=None, precision=None)

Bases: lightning_fabric.strategies.single_device.SingleDeviceStrategy

Strategy for training on a single TPU device.