SingleHPUStrategy¶
- class lightning.pytorch.strategies.SingleHPUStrategy(device='hpu', accelerator=None, checkpoint_io=None, precision_plugin=None)[source]¶
Bases:
lightning.pytorch.strategies.single_device.SingleDeviceStrategy
Strategy for training on single HPU device.
Warning
This is an experimental feature.
- optimizer_step(optimizer, closure, model=None, **kwargs)[source]¶
Performs the actual optimizer step.
- Parameters
- Return type