Python PyTorch Store.wait用法及代码示例

本文简要介绍python语言中 torch.distributed.Store.wait 的用法。

用法:

torch.distributed.Store.wait(*args, **kwargs)

参数：

keys(list) -在商店中设置它们之前等待的键列表。

参数：

• keys(list) -在商店中设置它们之前等待的键列表。

• timeout(时间增量) -在抛出异常之前等待添加键的时间。

重载函数。

1. wait(self: torch._C._distributed_c10d.Store, arg0: List[str]) -> 无

等待keys 中的每个键被添加到存储中。如果不是所有键都在timeout 之前设置(在存储初始化期间设置)，那么wait 将引发异常。

例子：

>>> import torch.distributed as dist
>>> from datetime import timedelta
>>> # Using TCPStore as an example, other store types can also be used
>>> store = dist.TCPStore("127.0.0.1", 0, 1, True, timedelta(seconds=30))
>>> # This will throw an exception after 30 seconds
>>> store.wait(["bad_key"])

2. wait(self: torch._C._distributed_c10d.Store, arg0: List[str], arg1: datetime.timedelta) -> None

等待 keys 中的每个键添加到存储中，如果提供的 timeout 未设置键，则引发异常。

例子：

>>> import torch.distributed as dist
>>> from datetime import timedelta
>>> # Using TCPStore as an example, other store types can also be used
>>> store = dist.TCPStore("127.0.0.1", 0, 1, True, timedelta(seconds=30))
>>> # This will throw an exception after 10 seconds
>>> store.wait(["bad_key"], timedelta(seconds=10))