SingleTPUPlugin¶
- class pytorch_lightning.plugins.training_type.SingleTPUPlugin(device, debug=False)[source]¶
Bases:
pytorch_lightning.plugins.training_type.single_device.SingleDevicePlugin
Plugin for training on a single TPU device.
- pre_dispatch()[source]¶
Hook to do something before the training/evaluation/prediction starts.
- Return type