SingleTPUStrategy¶
- class lightning.fabric.strategies.SingleTPUStrategy(device, accelerator=None, checkpoint_io=None, precision=None)[source]¶
Bases:
SingleDeviceStrategyStrategy for training on a single TPU device.
- Docs >
- lightning.fabric.strategies >
- SingleTPUStrategy
-
Shortcuts
To analyze traffic and optimize your experience, we serve cookies on this site. By clicking or navigating, you agree to allow our usage of cookies. Read PyTorch Lightning's Privacy Policy.
