otx.backend.native.strategies

Lightning strategy for single XPU device.

Classes

SingleXPUStrategy([device, accelerator, ...])

Strategy for training on single XPU device.

class otx.backend.native.strategies.SingleXPUStrategy(device: _DEVICE = 'xpu:0', accelerator: pl.accelerators.Accelerator | None = None, checkpoint_io: CheckpointIO | None = None, precision_plugin: PrecisionPlugin | None = None)

Bases: SingleDeviceStrategy

Strategy for training on single XPU device.