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