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