accelerator_prepare

pybind11_ke.config.accelerator_prepare(*args: List[Any]) → List[Any]

由于分布式并行依赖于 accelerate ，因此，需要利用 Accelerator 为分布式训练准备对象。

例子:

dataloader, model, accelerator = accelerator_prepare(
    dataloader,
    model
)

参数:

args (List[Any]) – pybind11_ke.data.KGEDataLoader 和 pybind11_ke.module.strategy.Strategy 。

返回:

包装好的对象列表和 Accelerator() 对象。

返回类型:

List[Any]