SingleDeviceXLAStrategy¶
- class lightning.fabric.strategies.SingleDeviceXLAStrategy(device, accelerator=None, checkpoint_io=None, precision=None)[source]¶
Bases:
SingleDeviceStrategy
Strategy for training on a single XLA device.
Bases: SingleDeviceStrategy
Strategy for training on a single XLA device.