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