Python PyTorch HashStore用法及代碼示例

本文簡要介紹python語言中 torch.distributed.HashStore 的用法。

用法:

class torch.distributed.HashStore

基於底層 hashmap 的線程安全存儲實現。此存儲可以在同一進程中使用(例如，由其他線程使用)，但不能跨進程使用。

例子：

>>> import torch.distributed as dist
>>> store = dist.HashStore()
>>> # store can be used from other threads
>>> # Use any of the store methods after initialization
>>> store.set("first_key", "first_value")