Python PyTorch FileStore用法及代码示例

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

用法:

class torch.distributed.FileStore

参数：

• file_name(str) -存储键值对的文件的路径

• world_size(int,可选的) -使用存储的进程总数。默认值为 -1(负值表示商店用户数量不固定)。

使用文件存储底层键值对的存储实现。

例子：

>>> import torch.distributed as dist
>>> store1 = dist.FileStore("/tmp/filestore", 2)
>>> store2 = dist.FileStore("/tmp/filestore", 2)
>>> # Use any of the store methods from either the client or server after initialization
>>> store1.set("first_key", "first_value")
>>> store2.get("first_key")