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))