ray.train.torch.xla.TorchXLAConfig#

class ray.train.torch.xla.TorchXLAConfig(backend: str | None = None, init_method: str = 'env', timeout_s: int = 1800, neuron_parallel_compile: bool = False)[源代码]#

基类：TorchConfig

torch XLA 设置的配置。更多信息请参见 https://pytorch.org/xla/release/1.13/index.html。目前，仅支持使用 xrt 运行时的 “neuron_cores” 加速器（AwsNeuronXLABackend）。

PublicAPI (alpha): 此API处于alpha阶段，可能在稳定之前发生变化。

方法

属性

`backend`
`backend_cls`
`init_method`
`neuron_parallel_compile`
`timeout_s`
`train_func_context`